Update app.py

app.py

@@ -76,6 +76,7 @@ def ollama_func(command):
         c1, c2 = command.split(" ")
     else:
         c1 = command
+        c2 = ""
     function_map = {
         "/init": init(),
         "/pull": ollama.pull(c2),
@@ -84,9 +85,9 @@ def ollama_func(command):
     }
     if c1 in function_map:
         function_map.get(c1)
-        return "", [("", f"Running {c1}")]
+        yield "Running..."
     else:
-        return "", [("", "No supported command.")]
+        yield "No supported command."
         
 @spaces.GPU()
 def launch():
@@ -100,7 +101,7 @@ def stream_chat(message: str, history: list, model: str, temperature: float, max
         ollama_func(message)
     else:
         if not INIT_SIGN:
-            return "", [(message, "Please initialize Ollama")]
+            yield "Please initialize Ollama"
         else:
             launch()
             conversation = []