devkunalnaik commited on
Commit
3373714
·
1 Parent(s): e5da9d1

Fix build: remove basicsr/gfpgan (Python 3.13 incompatible), use onnxruntime CPU

Browse files
Files changed (2) hide show
  1. processors/face_swap.py +4 -3
  2. 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=["CUDAExecutionProvider", "CPUExecutionProvider"],
67
  )
68
- self._app.prepare(ctx_id=0, det_size=(640, 640))
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=["CUDAExecutionProvider", "CPUExecutionProvider"],
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-gpu>=1.16.0
11
 
12
- # Face Enhancement
13
- gfpgan>=1.3.8
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