Update models.py

models.py

@@ -65,13 +65,13 @@ def load_face_analysis():
     print("Loading face analysis model...")
     try:
         face_app = FaceAnalysis(
-            name=FACE_DETECTION_CONFIG['model_name'],
-            root='./models/insightface',
-            providers=['CUDAExecutionProvider', 'CPUExecutionProvider']
+            name='antelopev2',
+            root='/data',
+            providers=['CPUExecutionProvider']
         )
         face_app.prepare(
-            ctx_id=FACE_DETECTION_CONFIG['ctx_id'], 
-            det_size=FACE_DETECTION_CONFIG['det_size']
+            ctx_id=0, 
+            det_size=(640, 640)
         )
         print("  [OK] Face analysis model loaded successfully")
         return face_app, True
@@ -84,7 +84,9 @@ def load_depth_detector():
     """Load Zoe Depth detector."""
     print("Loading Zoe Depth detector...")
     try:
-        zoe_depth = ZoeDetector.from_pretrained("lllyasviel/Annotators")
+        zoe_depth = ZoeDetector.from_pretrained(
+            "lllyasviel/Annotators"
+        )
         zoe_depth.to(device)
         print("  [OK] Zoe Depth loaded successfully")
         return zoe_depth, True