Update app.py

app.py

@@ -257,8 +257,8 @@ with gr.Blocks(theme=theme) as blk_demo:
     with gr.Row():
         with gr.Column():
             with gr.Box():
-                trg_in = gr.Image(type="pil", label='Target').style(height=300)
-                src_in = gr.Image(type="pil", label='Source').style(height=300)
+                trg_in = gr.Image(shape=None, type="pil", label='Target').style(height=300)
+                src_in = gr.Image(shape=None, type="pil", label='Source').style(height=300)
             with gr.Row():
                 b1 = gr.Button("Face Swap")
             with gr.Row():
@@ -281,7 +281,7 @@ with gr.Blocks(theme=theme) as blk_demo:
             with gr.Box():
                 ano_out = gr.Image(type="pil", label='Output').style(height=300)
 
-    b1.click(run_inference, inputs=[trg_in, src_in, def_in, mrg_in, chk_in], outputs=[ano_out])
+    b1.click(run_inference, inputs=[trg_in, src_in, def_in, mrg_in, chk_in], outputs=ano_out)
 """iface = gradio.Interface(run_inference,
                          [gradio.Image(shape=None, type="pil", label='Target'),
                           gradio.Image(shape=None, type="pil", label='Source'),