Update app.py

app.py

@@ -62,10 +62,15 @@ def generate(
     }
 
     def run_generation():
-        return pipe(input_text, **generate_kwargs)
+        try:
+            return pipe(input_text, **generate_kwargs)
+        except Exception as e:
+            gr.Error(f"Error in generation: {e}")
+            return []
 
     t = Thread(target=run_generation)
     t.start()
+    t.join()  # Ensure the thread completes before proceeding
 
     outputs = []
     for text in run_generation():