Spaces:

profaker
/

FaceSwapper

Runtime error

profaker commited on May 3, 2024

Commit

d1357fb

verified ·

1 Parent(s): e71138d

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]