Update app.py

app.py

@@ -127,7 +127,7 @@ def stream_chat(message: str, history: list, model: str, temperature: float, max
             },
         )
         print(response)
-        return response
+        return response['message']['content']
 
 
 
@@ -152,8 +152,8 @@ def main(message: str, history: list, model: str, temperature: float, max_new_to
                 top_k,
                 penalty             
             )
-            
-            return response['message']['content']
+            print(response)
+            yield response