Update app.py

app.py

@@ -153,13 +153,9 @@ def main(message: str, history: list, model: str, temperature: float, max_new_to
                 penalty             
             )
             
-            print(response)
+            yield response['message']['content']
+            
             
-            buffer = ""
-            for chunk in response:
-                buffer += chunk["message"]["content"]
-                yield buffer
-
 
 chatbot = gr.Chatbot(height=600, placeholder=DESCRIPTION)