Update app.py

app.py

@@ -136,7 +136,7 @@ async def stream_chat(message: str, history: list, model: str, temperature: floa
             yield buffer
 
 
-def main(message: str, history: list, model: str, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
+async def main(message: str, history: list, model: str, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
     if message.startswith("/"):
         resp = ollama_func(message)
         yield resp
@@ -147,7 +147,7 @@ def main(message: str, history: list, model: str, temperature: float, max_new_to
             if not process:
                 launch()
                
-            await response = stream_chat(
+            response = await stream_chat(
                 message,
                 history,
                 model,