Update app.py

app.py

@@ -98,10 +98,12 @@ def launch():
     OLLAMA_SERVICE_THREAD = threading.Thread(target=ollama_service_thread)
     OLLAMA_SERVICE_THREAD.start()
     print("Giving ollama serve a moment")
-    time.sleep(10)
+    time.sleep(5)
 
 @spaces.GPU()  
 async def stream_chat(message: str, history: list, model: str, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
+    if not process:
+        launch()
     print(f"message: {message}")
     conversation = []
     for prompt, answer in history:
@@ -143,10 +145,7 @@ async def main(message: str, history: list, model: str, temperature: float, max_
     else:
         if not INIT_SIGN:
             yield "Please initialize Ollama"
-        else:
-            if not process:
-                launch()
-               
+        else:         
             async for response in stream_chat(
                 message,
                 history,