Assistant startup fix

artitw · artitw · commit 415b7988f7b7 · 2024-09-22T22:34:25.000Z
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
   name="text2text",
-  version="1.5.0",
+  version="1.5.1",
   author="artitw",
   author_email="artitw@gmail.com",
   description="Text2Text: Crosslingual NLP/G toolkit",
diff --git a/text2text/assistant.py b/text2text/assistant.py
@@ -28,7 +28,7 @@ def __init__(self, **kwargs):
 
   def load_model(self):
     sub = subprocess.Popen(
-      f"ollama serve & ollama pull {self.model_name}", 
+      f"ollama serve & ollama pull {self.model_name} & ollama run {self.model_name}", 
       shell=True, 
       stdout=subprocess.PIPE
     )

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ def __init__(self, **kwargs):`
`28`	`28`
`29`	`29`	`def load_model(self):`
`30`	`30`	`sub = subprocess.Popen(`
`31`		`- f"ollama serve & ollama pull {self.model_name}",`
	`31`	`+ f"ollama serve & ollama pull {self.model_name} & ollama run {self.model_name}",`
`32`	`32`	`shell=True,`
`33`	`33`	`stdout=subprocess.PIPE`
`34`	`34`	`)`