Deeev2005 commited on
Commit
38bae98
·
verified ·
1 Parent(s): 752b61f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ os.environ["XDG_RUNTIME_DIR"] = "/tmp/runtime-dir"
10
  app = FaceAnalysis(name="buffalo_l", providers=["CUDAExecutionProvider","CPUExecutionProvider"])
11
  app.prepare(ctx_id=0, det_size=(1024,1024))
12
 
13
- onnx_model_path = "./inswapper_128.onnx" # <-- put model in Space files or download
14
  swap_model = get_model(onnx_model_path,
15
  providers=["CUDAExecutionProvider","CPUExecutionProvider"])
16
 
 
10
  app = FaceAnalysis(name="buffalo_l", providers=["CUDAExecutionProvider","CPUExecutionProvider"])
11
  app.prepare(ctx_id=0, det_size=(1024,1024))
12
 
13
+ onnx_model_path = "inswapper_128.onnx" # <-- put model in Space files or download
14
  swap_model = get_model(onnx_model_path,
15
  providers=["CUDAExecutionProvider","CPUExecutionProvider"])
16