Add model loading test function for better debugging

app.py

@@ -240,9 +240,63 @@ def detect_faces(image, model_type="mobilenet", confidence_threshold=0.5, nms_th
     except Exception as e:
         return None, f"Error: {str(e)}"
 
+# Simple test function to debug model loading
+def test_model_loading():
+    """Test model loading step by step"""
+    try:
+        print("=== Testing Model Loading ===")
+        
+        # Test basic imports
+        print("Testing RetinaFace import...")
+        test_cfg = {
+            'name': 'mobilenet0.25',
+            'min_sizes': [[16, 32], [64, 128], [256, 512]],
+            'steps': [8, 16, 32],
+            'variance': [0.1, 0.2],
+            'clip': False,
+            'pretrain': False,
+            'return_layers': {'stage1': 1, 'stage2': 2, 'stage3': 3},
+            'in_channel': 32,
+            'out_channel': 64
+        }
+        
+        print("Creating RetinaFace instance...")
+        model = RetinaFace(cfg=test_cfg, phase='test')
+        print(f"✅ Model created successfully: {type(model)}")
+        
+        print("Checking model file...")
+        if os.path.exists('mobilenet0.25_Final.pth'):
+            print("✅ Model file exists")
+            
+            print("Loading state dict...")
+            state_dict = torch.load('mobilenet0.25_Final.pth', map_location='cpu')
+            print(f"✅ State dict loaded, keys: {len(state_dict.keys())}")
+            
+            print("Loading state dict into model...")
+            model.load_state_dict(state_dict)
+            print("✅ State dict loaded successfully!")
+            
+            return True
+        else:
+            print("❌ Model file not found")
+            return False
+            
+    except Exception as e:
+        import traceback
+        print(f"❌ Test failed: {e}")
+        print(f"❌ Traceback: {traceback.format_exc()}")
+        return False
+
 # Load models on startup
 print("Loading RetinaFace models...")
-model_loaded = load_models()
+print("Running model loading test...")
+test_result = test_model_loading()
+if test_result:
+    print("Test passed, proceeding with full model loading...")
+    model_loaded = load_models()
+else:
+    print("Test failed, skipping model loading...")
+    model_loaded = False
 
 # Create simple Gradio interface
 def create_interface():