Update app.py

app.py

@@ -92,7 +92,7 @@ def infer(
     generator = torch.Generator(device=device).manual_seed(seed)
 
     result = pipe(
-        image=image_subject["composite"],
+        image=image_subject,
         prompt=prompt,
         # height=height,
         # width=width,
@@ -125,7 +125,7 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
         with gr.Row():
             with gr.Column():
                 with gr.Row():
-                    image_subject = gr.ImageEditor(label="input image", type="pil")
+                    image_subject = gr.Image(label="input image", type="pil")
                     image_background = gr.Image(label="background Image", type="pil", visible=False)
                 prompt =  gr.Textbox(label="prompt")
                 run_button = gr.Button("Fuse", variant="primary")