Update app.py

app.py

@@ -126,8 +126,8 @@ def stream_chat(message: str, history: list, model: str, temperature: float, max
                 "keep_alive": "60s",
             },
         )
-
-        yield response
+        print(response)
+        return response
 
 
 
@@ -153,7 +153,7 @@ def main(message: str, history: list, model: str, temperature: float, max_new_to
                 penalty             
             )
             
-            yield response['message']['content']
+            return response['message']['content']