Update app.py

app.py

@@ -72,10 +72,12 @@ with gr.Blocks() as app:
     gr.HTML(html_section_1)
     gr.HTML(html_section_2)
     gr.HTML(html_section_3)
-    gr.Interface(
-        fn=swap_face,
-        inputs=[gr.Image(), gr.Image(), gr.Checkbox(label="face_enhancer?", info="do face enhancer?")],
-        outputs="image"
-    )
+    source_img = gr.Image()
+    target_img = gr.Image()
+    face_enhancer = gr.Checkbox(label="face_enhancer?", info="do face enhancer?")
+    output_img = gr.Image()
+
+    btn = gr.Button("Submit")
+    btn.click(swap_face, inputs=[source_img, target_img, face_enhancer], outputs=output_img)
 
 app.launch()