Update app.py

app.py

@@ -180,21 +180,16 @@ def chat_tab():
         with sess.post(
             ENDPOINT_URL, headers=HEADERS, data=payload, stream=True
         ) as response:
-            print(response.content)
             if response.status_code == 200:
                 for chunk in response.iter_content(chunk_size=4):
-                    try:
-                        if chunk:
-                            decoded = chunk.decode("utf-8")
-                            print(decoded)
-                            full_output += decoded
-                            if full_output.endswith("User:"):
-                                yield full_output[:-5]
-                                # break
-                            else:
-                                yield full_output
-                    except:
-                        print(chunk)
+                    if chunk:
+                        decoded = chunk.decode("utf-8")
+                        full_output += decoded
+                        if full_output.endswith("User:"):
+                            yield full_output[:-5]
+                            break
+                        else:
+                            yield full_output
 
     with gr.Column():
         (