V1

app.py

@@ -22,6 +22,8 @@ import gradio as gr
 from PIL import Image
 from torchvision import transforms
 import traceback
+from configs.get_config import load_config
+from models import *
 
 # Add current directory to path
 if not os.getcwd() in sys.path:
@@ -40,8 +42,7 @@ except ImportError:
 
 # Check for custom models
 try:
-    from configs.get_config import load_config
-    from models import *
+    
     MODELS_IMPORTED = True
 except ImportError:
     print("Warning: Custom models couldn't be imported. Only damage detection will work.")
@@ -120,6 +121,9 @@ def setup_damage_detector(model_path, threshold=0.7):
         return None, None
     
     try:
+        print(f"Checking model path: {model_path}")
+        print(f"Model exists: {os.path.exists(model_path)}")
+        
         if model_path is None or not os.path.exists(model_path):
             print(f"Error: Damage model file not found at {model_path}")
             return None, None
@@ -131,8 +135,8 @@ def setup_damage_detector(model_path, threshold=0.7):
         cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = threshold
         
         # Use CPU if on Mac (MPS)
-        if hasattr(torch, 'backends') and hasattr(torch.backends, 'mps') and torch.backends.mps.is_available():
-            cfg.MODEL.DEVICE = "cpu"
+        cfg.MODEL.DEVICE = "cpu"
+        print("Forcing Detectron2 to use CPU")        
         
         predictor = DefaultPredictor(cfg)
         return predictor, cfg

requirements.txt

@@ -1,9 +1,10 @@
 torch>=2.0.0
 torchvision>=0.15.0
-opencv-python>=4.8.0
+opencv-python
 numpy>=1.24.0
 Pillow>=10.0.0
 gradio>=3.50.0
+python-box>=7.1.1
 fvcore>=0.1.5.post20221221; platform_system!="Darwin"
 iopath>=0.1.9; platform_system!="Darwin"
 pycocotools>=2.0.6; platform_system!="Darwin"