profaker commited on
Commit
6715af7
1 Parent(s): b18d1e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def Swap_Face(image1,image2):
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=onnxruntime.get_available_providers())
15
  face_enhancer = GFPGANer(model_path="GFPGANv1.4.pth", upscale=1,device="cuda")
16
 
17
  # Do the swap
 
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=['CUDAExecutionProvider'])
15
  face_enhancer = GFPGANer(model_path="GFPGANv1.4.pth", upscale=1,device="cuda")
16
 
17
  # Do the swap