Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,12 +7,13 @@ import onnxruntime
|
|
| 7 |
|
| 8 |
|
| 9 |
def Swap_Face(image1,image2):
|
| 10 |
-
|
|
|
|
| 11 |
app.prepare(ctx_id=0, det_size=(640, 640))
|
| 12 |
-
|
| 13 |
swapper = insightface.model_zoo.get_model("inswapper_128.onnx", download=False,
|
| 14 |
-
download_zip=False,providers=
|
| 15 |
-
face_enhancer = GFPGANer(model_path="GFPGANv1.4.pth", upscale=1
|
| 16 |
|
| 17 |
# Do the swap
|
| 18 |
face1 = app.get(image1)[0]
|
|
|
|
| 7 |
|
| 8 |
|
| 9 |
def Swap_Face(image1,image2):
|
| 10 |
+
providers = onnxruntime.get_available_providers()
|
| 11 |
+
app = FaceAnalysis(name='buffalo_l',providers=providers)
|
| 12 |
app.prepare(ctx_id=0, det_size=(640, 640))
|
| 13 |
+
|
| 14 |
swapper = insightface.model_zoo.get_model("inswapper_128.onnx", download=False,
|
| 15 |
+
download_zip=False,providers=providers)
|
| 16 |
+
face_enhancer = GFPGANer(model_path="GFPGANv1.4.pth", upscale=1)
|
| 17 |
|
| 18 |
# Do the swap
|
| 19 |
face1 = app.get(image1)[0]
|