diff --git a/src/components/Option/Models/AddOllamaModelModal.tsx b/src/components/Option/Models/AddOllamaModelModal.tsx index 7ca2972..255bc9c 100644 --- a/src/components/Option/Models/AddOllamaModelModal.tsx +++ b/src/components/Option/Models/AddOllamaModelModal.tsx @@ -20,6 +20,7 @@ export const AddOllamaModelModal: React.FC = ({ open, setOpen }) => { }) const pullModel = async (modelName: string) => { + modelName.replaceAll("ollama pull", "").replaceAll("ollama run", "").trim() notification.info({ message: t("manageModels.notification.pullModel"), description: t("manageModels.notification.pullModelDescription", { @@ -52,7 +53,7 @@ export const AddOllamaModelModal: React.FC = ({ open, setOpen }) => {