Update app.py

app.py

@@ -47,8 +47,9 @@ def safe_load_yolo_model(path):
 
 model_paths = {
     'YOLOv11': './data/yolo11n.pt',
-    'Crack & Pothole Detector': './data/best.pt',
-    'Toll gates': './data/best2.pt'
+    'Crack & Pothole Detector': './data/crack.pt',
+    'Toll gates': './data/tollgates.pt',
+    'Railway Bridges': './data/bridges.pt'
 }
 
 models = {name: safe_load_yolo_model(path).to("cuda" if torch.cuda.is_available() else "cpu") for name, path in model_paths.items()}
@@ -59,7 +60,8 @@ for name, model in models.items():
 model_colors = {
     'YOLOv11': (0, 255, 0),  # Green
     'Crack & Pothole Detector': (255, 0, 0),  # Red
-    'Toll gates': (0, 0, 255)  # Blue
+    'Toll gates': (0, 0, 255),  # Blue
+    'Railway Bridges': (0, 255, 255)  # Yellow
 }
 
 # Helper functions