pull model

app.py

@@ -1,9 +1,12 @@
+
+import os
+import threading
+import time
 import subprocess
+
 #from transformers import pipeline
 import ollama
 import gradio
-import os
-import threading
 
 OLLAMA = os.path.expanduser("~/ollama")
 
@@ -11,11 +14,17 @@ if not os.path.exists(OLLAMA):
     subprocess.run("curl -L https://ollama.com/download/ollama-linux-amd64 -o ~/ollama", shell=True)
     os.chmod(OLLAMA, 0o755)
 
+
 history = []
 
 def ollama_service_thread():
     subprocess.run("~/ollama serve", shell=True)
 
+print("Giving ollama serve a moment")
+time.sleep(10)
+subprocess.run("~/ollama pull tinydolphin:latest")
+
+
 OLLAMA_SERVICE_THREAD = threading.Thread(target=ollama_service_thread)
 OLLAMA_SERVICE_THREAD.start()