Update app.py

app.py

@@ -1,60 +1,48 @@
 import gradio as gr
 import numpy as np
-import os
-import cv2
-from PIL import Image, ImageDraw
+from PIL import Image
 import insightface
 from insightface.app import FaceAnalysis
 
-# Insightface model
-app = FaceAnalysis(name='buffalo_l')
+# ---- 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=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
+
+# ---- Main function ----
 def generate_image(src_img, dest_img):
-    # Convert to RGB
-    #src_img = src_img.convert(mode='RGB')
-    #dest_img = dest_img.convert(mode='RGB')
-
-    # Convert to array
-    #src_img_arr = np.asarray(src_img)
-    #dest_img_arr = np.asarray(dest_img)
-    src_img_arr = src_img
-    dest_img_arr = dest_img
-
-    # Face detection
-    src_faces = app.get(src_img_arr)
-    dest_faces = app.get(dest_img_arr)
-
-    # Initialize swapper
-    #swapper = insightface.model_zoo.get_model('./inswapper_128.onnx', download=False, download_zip=False)
-    #swapper = insightface.model_zoo.get_model('./inswapper_128.onnx')
+    # src_img, dest_img arrive as numpy arrays
+    src_faces = app.get(src_img)
+    dest_faces = app.get(dest_img)
 
-
-    # Swap face
-    res = dest_img_arr.copy()
+    res = dest_img.copy()
     for face in dest_faces:
-    	res = swapper.get(res, face, src_faces[0], paste_back=True)
-
-    # Convert to PIL image
-    final_image = Image.fromarray(np.uint8(res)).convert('RGB')
+        res = swapper.get(res, face, src_faces[0], paste_back=True)
 
-    return final_image 
+    return Image.fromarray(np.uint8(res)).convert("RGB")
 
-
-iface = gr.Interface(fn=generate_image, inputs=[
-    gr.Image(label="Source Image"),
-    gr.Image(label="Destination Image")], 
+# ---- Interface ----
+iface = gr.Interface(
+    fn=generate_image,
+    inputs=[gr.Image(label="Source Image"), gr.Image(label="Destination Image")],
     outputs=gr.Image(label="Final Image"),
-    theme=gr.themes.Base(primary_hue=gr.themes.colors.teal, font=["helvetica"]),
     examples=[["Images/kim.jpg", "Images/marilyn.jpg"]],
-    description=
-        """ # Face Swap
-        by <a href="https://www.tonyassi.com/" style="color: #97d8be;">Tony Assi</a>
-        
-        Try out [Video Face Swap](https://huggingface.co/spaces/tonyassi/video-face-swap). Please ❤️ this Space.
-
-        """
-    )
-iface.launch()
+    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()