Update src/faceswap.py
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 = []
|