Fix Assistant

artitw · Oct 7, 2024 · 39094fb · 39094fb
1 parent 64331b0
commit 39094fb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
   name="text2text",
-  version="1.6.5",
+  version="1.6.6",
   author="artitw",
   author_email="artitw@gmail.com",
   description="Text2Text: Crosslingual NLP/G toolkit",

diff --git a/text2text/assistant.py b/text2text/assistant.py
@@ -67,7 +67,7 @@ def load_model(self):
       response = requests.get("https://ollama.com/install.sh")
       install_script = response.text
       result = run_sh(install_script)
-      if "Install complete." not in result and "will run in CPU-only mode." not in result:
+      if result and "Install complete." not in result and "will run in CPU-only mode." not in result:
         raise Exception(result)
 
       self.ollama_serve_proc = subprocess.Popen(["ollama", "serve"])