catch error

app.py

@@ -16,26 +16,30 @@ def sanitize_prompt(prompt):
 
 @spaces.GPU(duration=180)
 def process_images(image, image2=None,prompt="a girl",inpaint_model="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0):
-    # I'm not sure when this happen
-    if not isinstance(image, dict):
-        if image2 == None:
-             print("empty mask")
-             return image
-        else:
-            image = dict({'background': image, 'layers': [image2]})
+    try:
+        # I'm not sure when this happen
+        if not isinstance(image, dict):
+            if image2 == None:
+                print("empty mask")
+                return image
+            else:
+                image = dict({'background': image, 'layers': [image2]})
 
-    if image2!=None:
-         #print("use image2")
-         mask = image2
-    else:
-         if len(image['layers']) == 0:
-              print("empty mask")
-              return image
-         print("use layer")
-         mask = image['layers'][0]
+        if image2!=None:
+            #print("use image2")
+            mask = image2
+        else:
+            if len(image['layers']) == 0:
+                print("empty mask")
+                return image
+            print("use layer")
+            mask = image['layers'][0]
 
 
-    output = flux1_inpaint.process_image(image["background"],mask,prompt,inpaint_model,strength,seed)
+        output = flux1_img2img.process_image(image["background"],mask,prompt,inpaint_model,strength,seed)
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        gr.Error(e)
         
     return output
     
@@ -77,7 +81,7 @@ with demo_blocks as demo:
                         with gr.Row():
                             prompt = gr.Textbox(label="Prompt",value="a eyes closed girl,shut eyes",placeholder="Your prompt (what you want in place of what is erased)", elem_id="prompt")
                             
-                    btn = gr.Button("Inpaint", elem_id="run_button",variant="primary")
+                    btn = gr.Button("Img2Img", elem_id="run_button",variant="primary")
                     
                     image_mask = gr.Image(sources=['upload','clipboard'],  elem_id="mask_upload", type="pil", label="Mask_Upload",height=400, value=None)
                     with gr.Accordion(label="Advanced Settings", open=False):