Update app.py

app.py

@@ -1,14 +1,19 @@
 import cv2
 import numpy as np
 import gradio as gr
+
+import torch
 import torch.serialization
-torch.serialization.add_safe_globals(['ultralytics.nn.tasks.DetectionModel'])
+import ultralytics.nn.tasks as tasks
+torch.serialization.add_safe_globals({tasks.DetectionModel})
 
 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²