Spaces:
Running
Running
Commit ·
3373714
1
Parent(s): e5da9d1
Fix build: remove basicsr/gfpgan (Python 3.13 incompatible), use onnxruntime CPU
Browse files- processors/face_swap.py +4 -3
- requirements.txt +3 -6
processors/face_swap.py
CHANGED
|
@@ -61,17 +61,18 @@ class FaceSwapper:
|
|
| 61 |
from insightface.app import FaceAnalysis
|
| 62 |
|
| 63 |
# Face analysis (buffalo_l auto-downloads on first run)
|
|
|
|
| 64 |
self._app = FaceAnalysis(
|
| 65 |
name="buffalo_l",
|
| 66 |
-
providers=["
|
| 67 |
)
|
| 68 |
-
self._app.prepare(ctx_id=
|
| 69 |
|
| 70 |
# inswapper model
|
| 71 |
_download_file(INSWAPPER_URL, INSWAPPER_PATH)
|
| 72 |
self._swapper = insightface.model_zoo.get_model(
|
| 73 |
str(INSWAPPER_PATH),
|
| 74 |
-
providers=["
|
| 75 |
)
|
| 76 |
|
| 77 |
self._ready = True
|
|
|
|
| 61 |
from insightface.app import FaceAnalysis
|
| 62 |
|
| 63 |
# Face analysis (buffalo_l auto-downloads on first run)
|
| 64 |
+
# Initialize face analysis (CPU-only for free HF Spaces tier)
|
| 65 |
self._app = FaceAnalysis(
|
| 66 |
name="buffalo_l",
|
| 67 |
+
providers=["CPUExecutionProvider"],
|
| 68 |
)
|
| 69 |
+
self._app.prepare(ctx_id=-1, det_size=(640, 640))
|
| 70 |
|
| 71 |
# inswapper model
|
| 72 |
_download_file(INSWAPPER_URL, INSWAPPER_PATH)
|
| 73 |
self._swapper = insightface.model_zoo.get_model(
|
| 74 |
str(INSWAPPER_PATH),
|
| 75 |
+
providers=["CPUExecutionProvider"],
|
| 76 |
)
|
| 77 |
|
| 78 |
self._ready = True
|
requirements.txt
CHANGED
|
@@ -7,13 +7,10 @@ torchvision>=0.16.0
|
|
| 7 |
|
| 8 |
# Face Swap
|
| 9 |
insightface>=0.7.3
|
| 10 |
-
onnxruntime
|
| 11 |
|
| 12 |
-
# Face Enhancement
|
| 13 |
-
gfpgan
|
| 14 |
-
facexlib>=0.3.0
|
| 15 |
-
realesrgan>=0.3.0
|
| 16 |
-
basicsr>=1.4.2
|
| 17 |
|
| 18 |
# Body Segmentation
|
| 19 |
rembg>=2.0.50
|
|
|
|
| 7 |
|
| 8 |
# Face Swap
|
| 9 |
insightface>=0.7.3
|
| 10 |
+
onnxruntime>=1.16.0
|
| 11 |
|
| 12 |
+
# Face Enhancement (optional - installed separately if GPU available)
|
| 13 |
+
# gfpgan, facexlib, realesrgan, basicsr excluded: incompatible with Python 3.13
|
|
|
|
|
|
|
|
|
|
| 14 |
|
| 15 |
# Body Segmentation
|
| 16 |
rembg>=2.0.50
|