Update app.py

app.py

@@ -4,6 +4,8 @@ import torch
 from datetime import datetime
 from ultralytics import YOLO
 
+torch.serialization.add_safe_globals([torch.nn.Module, 'ultralytics.nn.tasks.DetectionModel'])
+
 # Load YOLOv8 model (trained on construction dataset)
 model = YOLO('yolov8n.pt')  # Path to pre-trained model on construction dataset