Update app.py

app.py

@@ -1,12 +1,14 @@
 import cv2
 import numpy as np
 import gradio as gr
-from ultralyticsplus import YOLO
+import torch.serialization
+torch.serialization.add_safe_globals(['ultralytics.nn.tasks.DetectionModel'])
 
-# Load YOLO model
+from ultralyticsplus import YOLO
 model = YOLO("keremberke/yolov8n-pothole-segmentation")
 model.overrides['conf'] = 0.25
 
+
 # Constants
 SCALE = 0.005  # m² per pixel²
 COST_PER_M2 = 1000  # ₹ per m²