Update app.py

app.py

@@ -66,19 +66,12 @@ def swap_face(source_file, target_file, doFaceEnhancer):
 app = gr.Blocks()
 
 with app:
-    source_image = gr.Image(label="Source Image")
-    target_image = gr.Image(label="Target Image")
-    face_enhancer = gr.Checkbox(label="Apply Enhancer?", info="Face Enhancer")
-    output_image = gr.Image(label="Output Image")
-
-    swap_button = gr.Button("Swap Faces")
-    swap_button.click(
+    gr.Interface(
         fn=swap_face,
-        inputs=[source_image, target_image, face_enhancer],
-        outputs=output_image,
+        inputs=[gr.Image(label="Source Image"), gr.Image(label="Target Image"), gr.Checkbox(label="Apply Enhancer?", info="Face Enhancer")],
+        outputs=gr.Image(label="Output Image"),
+        description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",
+        theme="NoCrypt/miku"
     )
 
-app.launch(
-    description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",
-    theme="NoCrypt/miku"
-)
+app.launch()