Update app.py

app.py

@@ -6,7 +6,9 @@ import subprocess
 import spaces
 
 OLLAMA = os.path.expanduser("~/ollama")
-process = ""
+process = None
+OLLAMA_SERVICE_THREAD = None
+
 
 if not os.path.exists(OLLAMA):
     subprocess.run("curl -L https://ollama.com/download/ollama-linux-amd64 -o ~/ollama", shell=True)
@@ -17,13 +19,14 @@ def ollama_service_thread():
     process = subprocess.Popen("~/ollama serve", shell=True, preexec_fn=os.setsid)
     process.wait()
     
-OLLAMA_SERVICE_THREAD = threading.Thread(target=ollama_service_thread)
-# OLLAMA_SERVICE_THREAD.start()
-
 def terminate():
+    global process, OLLAMA_SERVICE_THREAD
     if process:
         os.killpg(os.getpgid(process.pid), signal.SIGTERM)
-    OLLAMA_SERVICE_THREAD.join()
+    if OLLAMA_SERVICE_THREAD:
+        OLLAMA_SERVICE_THREAD.join()
+    process = None
+    OLLAMA_SERVICE_THREAD = None
 
 # Uncomment and modify the model to what you want locally
 # model = "moondream" 
@@ -65,6 +68,8 @@ h3 {
 INIT_SIGN = ""
 
 def init():
+    global OLLAMA_SERVICE_THREAD
+    OLLAMA_SERVICE_THREAD = threading.Thread(target=ollama_service_thread)
     OLLAMA_SERVICE_THREAD.start()
     print("Giving ollama serve a moment")
     time.sleep(10)