Update model.py

model.py

@@ -29,7 +29,6 @@ class ModelHandler:
         """
         Load face analysis model. 
         Downloads from HF Hub to the path insightface expects.
-        Forces CPU to avoid ZeroGPU initialization errors.
         """
         print("Loading face analysis model...")
         
@@ -50,7 +49,7 @@ class ModelHandler:
             self.app = FaceAnalysis(
                 name=Config.ANTELOPEV2_NAME,
                 root=Config.ANTELOPEV2_ROOT, 
-                providers=['CPUExecutionProvider'] 
+                providers=['CUDAExecutionProvider', 'CPUExecutionProvider'] 
             )
             self.app.prepare(ctx_id=0, det_size=(640, 640))
             print(f"  [OK] Face analysis model loaded successfully.")
@@ -113,7 +112,7 @@ class ModelHandler:
         # --- NEW: Enable xFormers ---
         try:
             self.pipeline.enable_xformers_memory_efficient_attention()
-            print("  [OK] xFormers memory efficient attention enabled.")
+            print("  [OK] xFormmrs memory efficient attention enabled.")
         except Exception as e:
             print(f"  [WARNING] Failed to enable xFormers: {e}")
         # --- END NEW ---
@@ -160,7 +159,6 @@ class ModelHandler:
         
         print("--- All models loaded successfully ---")
 
-    # --- MODIFIED: Renamed function and changed return value ---
     def get_face_info(self, image):
         """Extracts the largest face, returns insightface result object."""
         if not self.face_analysis_loaded:
@@ -174,7 +172,9 @@ class ModelHandler:
                 return None
                 
             # Sort by size (width * height) to find the main character
-            faces = sorted(faces, key=lambda x: (x['bbox'][2]-x['bbox'][0])*(x['bbox'][3]-x['bbox'][1]), reverse=True)
+            # --- MODIFIED: Fixed typo x['bbox[0]) -> x['bbox'][0]) ---
+            faces = sorted(faces, key=lambda x: (x['bbox'][2]-x['bbox'][0])*(x['bbox'][3]-x'bbox'[1]), reverse=True)
+            # --- END MODIFIED ---
             
             # Return the largest face info
             return faces[0]