Update app.py

app.py

@@ -11,10 +11,9 @@ def safe_load_yolo_model(path):
 # Dictionary of model paths
 model_paths = {
     'YOLOv11': './data/yolo11n.pt',
-    'Crack & Pothole Detector': './data/best2.pt',
-    'Bridge Inspector': './data/bridge.pt',
-    'Road Surface Detector': './data/road.pt',
-    'Pipe Detector': './data/pipe.pt'
+    'Crack & Pothole Detector': './data/best.pt',
+    'Toll gates': './data/best2.pt'
+    
 }
 
 # Load models into memory
@@ -24,9 +23,7 @@ models = {name: safe_load_yolo_model(path) for name, path in model_paths.items()
 model_colors = {
     'YOLOv11': (0, 255, 0),
     'Crack & Pothole Detector': (255, 0, 0),
-    'Bridge Inspector': (0, 0, 255),
-    'Road Surface Detector': (255, 255, 0),
-    'Pipe Detector': (255, 0, 255)
+    'Toll gates': (0, 0, 255)
 }
 
 def process_video(video, selected_model):