bugfix catching errors

app.py

@@ -60,19 +60,19 @@ def generate(
 
     try:
         stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
+        output = ""
+
+        for response in stream:
+            output += response.token.text
+    
+            for stop_str in STOP_SEQUENCES:
+                if output.endswith(stop_str):
+                    output = output[:-len(stop_str)]
+                    output = output.rstrip()
+                    yield output
+            yield output
     except Exception as e:
         raise gr.Error(f"Error while generating: {e}")
-    output = ""
-
-    for response in stream:
-        output += response.token.text
-
-        for stop_str in STOP_SEQUENCES:
-            if output.endswith(stop_str):
-                output = output[:-len(stop_str)]
-                output = output.rstrip()
-                yield output
-        yield output
     return output