Update app.py

app.py

@@ -34,6 +34,27 @@ def predict(image_in_video, image_in_img):
     image = image_in_video or image_in_img
     return image
 
+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')
+
+img = ins_get_image('t1')
+faces = app.get(img)
+
+source_face = faces[0]
+bbox = source_face['bbox']
+bbox = [int(b) for b in bbox]
+
+source_face = faces[0]
+bbox = source_face['bbox']
+bbox = [int(b) for b in bbox]
+
+res = img.copy()
+for face in faces:
+  res = swapper.get(res, face, source_face, paste_back=True)
+
+    
+
 with gr.Blocks() as blocks:
     gr.Markdown("### Capture Image Using WebCam or Upload")