sophiemaw commited on
Commit
28f714e
·
verified ·
1 Parent(s): 9f7a847

Update src/faceswap.py

Browse files
Files changed (1) hide show
  1. src/faceswap.py +7 -1
src/faceswap.py CHANGED
@@ -171,7 +171,13 @@ def face_swap_multiple_identities(
171
  embedding_model = get_model("/app/src/w600k_r50.onnx", providers=["CPUExecutionProvider"])
172
  embedding_model.prepare(ctx_id=0)
173
  face_analyser.models["embedding"] = embedding_model
174
- swapper = get_model(inswapper_model_path, providers=["CPUExecutionProvider"])
 
 
 
 
 
 
175
 
176
  # Prepare source faces
177
  src_faces = []
 
171
  embedding_model = get_model("/app/src/w600k_r50.onnx", providers=["CPUExecutionProvider"])
172
  embedding_model.prepare(ctx_id=0)
173
  face_analyser.models["embedding"] = embedding_model
174
+
175
+ # Load swapper model
176
+ model_path = hf_hub_download(
177
+ repo_id="sophiemaw/inswapper",
178
+ filename="inswapper_128.onnx"
179
+ )
180
+ swapper = get_model(model_path, providers=["CPUExecutionProvider"])
181
 
182
  # Prepare source faces
183
  src_faces = []