fix example

app.py

@@ -28,26 +28,9 @@ def sanitize_prompt(prompt):
 
 
 @spaces.GPU(duration=120)
-def process_images(image, image2=None,prompt="a girl",strength=0.75,seed=0,inference_step=4,progress=gr.Progress(track_tqdm=True)):
+def process_images(image,prompt="a girl",strength=0.75,seed=0,inference_step=4,progress=gr.Progress(track_tqdm=True)):
     print("start process_images")
     progress(0, desc="Starting")
-    # 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]
 
 
     def process_img2img(image,prompt="a person",strength=0.75,seed=0,num_inference_steps=4):
@@ -67,7 +50,7 @@ def process_images(image, image2=None,prompt="a girl",strength=0.75,seed=0,infer
         # TODO support mask
         return output.images[0]
 
-    output = process_img2img(image["background"],mask,prompt,strength,seed,inference_step)
+    output = process_img2img(image,prompt,strength,seed,inference_step)
    
     print("end process_images")
     return output