Update app.py

app.py

@@ -69,9 +69,16 @@ def generate(
     t.start()
 
     outputs = []
-    for text in streamer:
-        outputs.append(text)
-        yield "".join(outputs)
+    try:
+        for text in streamer:
+            outputs.append(text)
+            yield "".join(outputs)
+    except Exception as e:
+        yield f"An error occurred during generation: {str(e)}"
+    finally:
+        if not outputs:
+            yield ""
+
 
 
 chat_interface = gr.ChatInterface(