Upload 3 files

app.py

@@ -0,0 +1,49 @@
+import gradio as gr
+import cv2
+import numpy as np
+from PIL import Image
+import insightface
+from insightface.app import FaceAnalysis
+
+# Load face detector and swapper
+app = FaceAnalysis(name='buffalo_l')
+app.prepare(ctx_id=0, det_size=(640, 640))
+swapper = insightface.model_zoo.get_model('inswapper_128.onnx')
+
+def to_numpy(img):
+    if isinstance(img, Image.Image):
+        return np.array(img.convert("RGB"))
+    return img
+
+def swap_faces(source_img, dest_img, source_index=1, dest_index=1):
+    source_np = to_numpy(source_img)
+    dest_np = to_numpy(dest_img)
+
+    # Detect faces
+    source_faces = sorted(app.get(source_np), key=lambda x: x.bbox[0])
+    dest_faces = sorted(app.get(dest_np), key=lambda x: x.bbox[0])
+
+    if len(source_faces) < source_index or len(dest_faces) < dest_index:
+        raise gr.Error("Face index out of range!")
+
+    source_face = source_faces[source_index - 1]
+    dest_face = dest_faces[dest_index - 1]
+
+    # Swap
+    result = swapper.get(dest_np, dest_face, source_face, paste_back=True)
+    result_rgb = cv2.cvtColor(result, cv2.COLOR_BGR2RGB)
+    return Image.fromarray(result_rgb)
+
+# Gradio app
+gr.Interface(
+    fn=swap_faces,
+    inputs=[
+        gr.Image(label="Source Image"),
+        gr.Image(label="Target Image"),
+        gr.Number(value=1, label="Source Face Index"),
+        gr.Number(value=1, label="Target Face Index"),
+    ],
+    outputs=gr.Image(label="Swapped Output"),
+    title="Face Swap with InsightFace",
+    description="Swap faces using InsightFace and ONNX model (inswapper_128.onnx)."
+).launch()

inswapper_128.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e4a3f08c753cb72d04e10aa0f7dbe3deebbf39567d4ead6dce08e98aa49e16af
+size 554253681

requirements.txt

@@ -0,0 +1,5 @@
+gradio
+insightface>=0.7
+onnxruntime
+opencv-python
+Pillow