starting point

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,12 @@
+---
+title: Face Swap
+emoji: 👩🏻‍🦰
+colorFrom: green
+colorTo: indigo
+sdk: gradio
+sdk_version: 3.41.2
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,51 @@
+import gradio as gr
+import insightface
+from insightface.app import FaceAnalysis
+
+wellcomingMessage = """
+    <h1>Face Swapping</h1>
+    <p>If you like this app, plase take a look at my <a href="https://www.meetup.com/tech-web3-enthusiasts-united-insightful-conversations/" target="_blank">Meetup Group</a>! There will be more interesting apps and events soon.</p>
+    <p>Happy <span style="font-size:500%;color:red;">&hearts;</span> coding!</p>
+"""
+
+assert insightface.__version__>='0.7'
+
+app = FaceAnalysis(name='buffalo_l')
+app.prepare(ctx_id=0, det_size=(640, 640))
+swapper = insightface.model_zoo.get_model('inswapper_128.onnx', download=True, download_zip=True)
+
+def swap_faces(faceSource, sourceFaceId, faceDestination, destFaceId):
+    faces = app.get(faceSource)
+    faces = sorted(faces, key = lambda x : x.bbox[0])
+    if len(faces) < sourceFaceId or sourceFaceId < 1:
+        raise gr.Error(f"Source image only contains {len(faces)} faces, but you requested face {sourceFaceId}")
+        
+    source_face = faces[sourceFaceId-1]
+
+    res_faces = app.get(faceDestination)
+    res_faces = sorted(res_faces, key = lambda x : x.bbox[0])
+    if len(res_faces) < destFaceId or destFaceId < 1:
+        raise gr.Error(f"Destination image only contains {len(res_faces)} faces, but you requested face {destFaceId}")
+    res_face = res_faces[destFaceId-1]
+
+    result = swapper.get(faceDestination, res_face, source_face, paste_back=True)
+
+    # for face in faces:
+    #     res = swapper.get(res, face, source_face, paste_back=True)
+    # cv2.imwrite("./t1_swapped.jpg", res)
+    return result
+
+gr.Interface(swap_faces, 
+    [
+        gr.Image(), 
+        gr.Number(precision=0, value=1, info='face position (from left, starting at 1)'), 
+        gr.Image(), 
+        gr.Number(precision=0, value=1, info='face position (from left, starting at 1)')
+    ],
+     gr.Image(),
+     description=wellcomingMessage,
+     examples=[
+            ['./Images/kim.jpg', 1, './Images/marilyn.jpg', 1],
+            ['./Images/friends.jpg', 2, './Images/friends.jpg', 1],
+        ],
+).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,2 @@
+insightface
+onnxruntime