Spaces:
Runtime error
Runtime error
Update models.py
Browse files
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 |
-
|
| 64 |
-
face_app = FaceAnalysis(
|
| 65 |
-
|
| 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():
|