Update app.py

app.py

@@ -25,6 +25,33 @@ app = FaceAnalysis(name='buffalo_l')
 app.prepare(ctx_id=0, det_size=(640, 640))
 
 
+def face_swap(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)
+
+    # 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)
+
+    # Swap face
+    res = dest_img_arr.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')
+
+    return final_image 
+
 
 def greet(description,color,features,occasion,type_):