Update app.py

app.py

@@ -84,7 +84,7 @@ def evf_sam_mask(image, prompt):
 
 @spaces.GPU(duration=150)
 def process(
-    input_image_editor: dict,
+    input_image: dict,
     input_text: str,
     inpaint_text: str,
     seed_slicer: int,
@@ -97,15 +97,15 @@ def process(
         gr.Info("Please enter a text prompt.")
         return None
 
-    image = input_image_editor['background']
+    #image = input_image_editor['background']
     #mask = input_image_editor['layers'][0]
-    print(f"type of image: {type(image)}")
-    mask = evf_sam_mask(image, input_text)
+    print(f"type of image: {type(input_image)}")
+    mask = evf_sam_mask(input_image, input_text)
     print(f"type of mask: {type(mask)}")
     print(f"inpaint_text: {inpaint_text}")
     print(f"input_text: {input_text}")
     
-    if not image:
+    if not input_image:
         gr.Info("Please upload an image.")
         return None
 
@@ -113,8 +113,8 @@ def process(
         gr.Info("Please draw a mask on the image.")
         return None
 
-    width, height = resize_image_dimensions(original_resolution_wh=image.size)
-    resized_image = image.resize((width, height), Image.LANCZOS)
+    width, height = resize_image_dimensions(original_resolution_wh=input_image.size)
+    resized_image = input_image.resize((width, height), Image.LANCZOS)
     resized_mask = mask.resize((width, height), Image.NEAREST)
 
     if randomize_seed_checkbox:
@@ -138,7 +138,7 @@ with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN)
     with gr.Row():
         with gr.Column():
-            input_image_editor_component = gr.ImageEditor(
+            input_image = gr.ImageEditor( #input_image_editor_component
                 label='Image',
                 type='pil',
                 sources=["upload", "webcam"],
@@ -202,7 +202,7 @@ with gr.Blocks() as demo:
     submit_button_component.click(
         fn=process,
         inputs=[
-            input_image_editor_component,
+            input_image, #input_image_editor_component,
             input_text_component,
             inpaint_text_component,
             seed_slicer_component,