Spaces:

Vignesh455
/

FaceSwap

Runtime error

Vignesh455 commited on May 3, 2024

Commit

cc1e6ed

verified ·

1 Parent(s): 8a075f7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ def Swap_Face(image1,image2):
     swapper = insightface.model_zoo.get_model("inswapper_128.onnx", download=False,
                                               download_zip=False,providers=onnxruntime.get_available_providers())
-    face_enhancer = GFPGANer(model_path="GFPGANv1.4.pth", upscale=1)
     # Do the swap
     face1 = app.get(image1)[0]

     swapper = insightface.model_zoo.get_model("inswapper_128.onnx", download=False,
                                               download_zip=False,providers=onnxruntime.get_available_providers())
+    face_enhancer = GFPGANer(model_path="GFPGANv1.4.pth", upscale=1,device="cuda")
     # Do the swap
     face1 = app.get(image1)[0]