Spaces:

sanchitnanda03
/

pothole

Runtime error

sanchitnanda03 commited on Jul 12

Commit

3eb1370

verified ·

1 Parent(s): d08173c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 import cv2
 import numpy as np
 import gradio as gr
+import torch.serialization
+torch.serialization.add_safe_globals(['ultralytics.nn.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²