Update app.py

app.py

@@ -85,9 +85,9 @@ def ollama_func(command):
     }
     if c1 in function_map:
         function_map.get(c1)
-        yield "Running..."
+        return "Running..."
     else:
-        yield "No supported command."
+        return "No supported command."
         
 @spaces.GPU()
 def launch():
@@ -98,7 +98,8 @@ def launch():
 def stream_chat(message: str, history: list, model: str, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
     print(f"message: {message}")
     if message.startswith("/"):
-        ollama_func(message)
+        resp = ollama_func(message)
+        yield resp
     else:
         if not INIT_SIGN:
             yield "Please initialize Ollama"