primerz commited on
Commit
fae896d
·
verified ·
1 Parent(s): be008fd

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +11 -5
models.py CHANGED
@@ -59,11 +59,17 @@ def download_model_with_retry(repo_id, filename, max_retries=None):
59
 
60
 
61
  def load_face_analysis():
62
-
63
- antelope_download = snapshot_download(repo_id="DIAMONIK7777/antelopev2", local_dir="/data/models/antelopev2")
64
- face_app = FaceAnalysis(name='antelopev2', root='/data', providers=['CPUExecutionProvider'])
65
- face_app.prepare(ctx_id=0, det_size=(640, 640))
66
-
 
 
 
 
 
 
67
  return face_app, True
68
 
69
  def load_depth_detector():
 
59
 
60
 
61
  def load_face_analysis():
62
+ """Load face analysis model with intelligent provider selection."""
63
+ print("Loading face analysis model...")
64
+ face_app = FaceAnalysis(
65
+ name=FACE_DETECTION_CONFIG['model_name'],
66
+ root='/data',
67
+ providers=['CPUExecutionProvider']
68
+ )
69
+ face_app.prepare(
70
+ ctx_id=ctx_id,
71
+ det_size=FACE_DETECTION_CONFIG['det_size']
72
+ )
73
  return face_app, True
74
 
75
  def load_depth_detector():