Update app.py

app.py

@@ -416,7 +416,6 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
             if randomize_seed:
                 seed = random.randint(0, MAX_SEED)
 
-
         # Generate image
         if image_input is not None:
             final_image = generate_image_to_image(prompt_mash, image_input, image_strength, steps, cfg_scale, width, height, seed)
@@ -434,13 +433,9 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
             raise gr.Error("Failed to generate image")
         
         return final_image, seed, gr.update(visible=False)
-    except spaces.zero.gradio.HTMLError as e:
-        if "You have exceeded your Pro GPU quota" in str(e):
-            return gr.Error("GPU quota exceeded. Please try again later."), seed, gr.update(visible=False)
-        raise
     except Exception as e:
         print(f"Error in run_lora: {str(e)}")
-        return gr.Error(f"An error occurred: {str(e)}"), seed, gr.update(visible=False)
+        return gr.Error(str(e)), seed, gr.update(visible=False)
 
 run_lora.zerogpu = True
 
@@ -497,7 +492,8 @@ def update_history(new_image, history):
     """Updates the history gallery with the new image."""
     if history is None:
         history = []
-    history.insert(0, new_image)
+    if not isinstance(new_image, gr.Error):
+        history.insert(0, new_image)
     return history
 
 css = '''