Update app.py

app.py

@@ -4,45 +4,38 @@ from PIL import Image
 import insightface
 from insightface.app import FaceAnalysis
 
-# ---- Load models once ----
+# ---------- Load models once ----------
 app = FaceAnalysis(name="buffalo_l")
 app.prepare(ctx_id=0, det_size=(640, 640))
-#swapper = insightface.model_zoo.get_model("./inswapper_128.onnx",download=False, download_zip=False)
-swapper = insightface.model_zoo.get_model('inswapper_128.onnx', download=True, download_zip=True)
+swapper = insightface.model_zoo.get_model("inswapper_128.onnx",
+                                          download=False, download_zip=False)
 
-# ---- Gradio <4.39 monkey-patch for bool schema bug ----
-import gradio_client.utils as gu
-if hasattr(gu, "get_type"):
-    _old_get_type = gu.get_type
-    def _safe_get_type(schema):
-        if isinstance(schema, bool):   # <-- avoid "bool is not iterable"
-            return "bool"
-        return _old_get_type(schema)
-    gu.get_type = _safe_get_type
+# ---------- HTML description ----------
+description_html = """
+    <h1>Face Swap</h1>
+    <p>by <a href="https://www.tonyassi.com/" target="_blank" style="color:#97d8be;">Tony Assi</a></p>
+    <p>Try out <a href="https://huggingface.co/spaces/tonyassi/video-face-swap" target="_blank">Video Face Swap</a> ❤️</p>
+"""
 
-# ---- Main function ----
-def generate_image(src_img, dest_img):
-    # src_img, dest_img arrive as numpy arrays
+# ---------- Swap logic ----------
+def swap_faces(src_img, dest_img):
     src_faces = app.get(src_img)
     dest_faces = app.get(dest_img)
 
     res = dest_img.copy()
-    for face in dest_faces:
-        res = swapper.get(res, face, src_faces[0], paste_back=True)
+    if src_faces and dest_faces:
+        res = swapper.get(res, dest_faces[0], src_faces[0], paste_back=True)
 
     return Image.fromarray(np.uint8(res)).convert("RGB")
 
-# ---- Interface ----
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=[gr.Image(label="Source Image"), gr.Image(label="Destination Image")],
-    outputs=gr.Image(label="Final Image"),
-    examples=[["Images/kim.jpg", "Images/marilyn.jpg"]],
-    description="""
-    <h1>Face Swap</h1>
-    by <a href="https://www.tonyassi.com/" target="_blank" style="color:#97d8be;">Tony Assi</a><br>
-    Try out <a href="https://huggingface.co/spaces/tonyassi/video-face-swap">Video Face Swap</a> ❤️
-    """
-)
-
-iface.launch()
+# ---------- Interface ----------
+gr.Interface(
+    fn=swap_faces,
+    inputs=[gr.Image(), gr.Image()],
+    outputs=gr.Image(),
+    description=description_html,
+    examples=[
+        ['./Images/kim.jpg', './Images/marilyn.jpg'],
+        ['./Images/friends.jpg', './Images/friends.jpg']
+    ],
+).launch()