Update app.py

app.py

@@ -72,7 +72,10 @@ def init():
     INIT_SIGN = "FINISHED"
 
 def ollama_func(command):
-    c1, c2 = command.split(" ")
+    if " " in command:
+        c1, c2 = command.split(" ")
+    else:
+        c1 = command
     function_map = {
         "/init": init(),
         "/pull": ollama.pull(c2),
@@ -81,8 +84,9 @@ def ollama_func(command):
     }
     if c1 in function_map:
         function_map[c1]
+        return [f"Running {function_map[c1]}"]
     else:
-        print("No supported command.")
+        return ["No supported command."]
         
 @spaces.GPU()
 def launch():
@@ -96,7 +100,7 @@ def stream_chat(message: str, history: list, model: str, temperature: float, max
         ollama_func(message)
     else:
         if not INIT_SIGN:
-            return "Please Enter /init to initialize Ollama"
+            return ["Please Enter /init to initialize Ollama"]
         else:
             launch()
             conversation = []