Update app.py

app.py

@@ -4,19 +4,20 @@ import gradio as gr
 import easyocr
 import numpy as np
 from PIL import Image
+import ultralytics.nn.tasks as tasks
 
-# Load YOLO model (trusted official weights)
-WEIGHTS_PATH = "yolov8n.pt"  # Make sure this file exists or is downloaded in HF Space
+# ✅ Allowlist the YOLOv8 DetectionModel class so PyTorch 2.6 can unpickle it safely
+torch.serialization.add_safe_globals([tasks.DetectionModel])
 
-# Force PyTorch to load with weights_only=False
+# Load YOLO model (trusted official weights from Ultralytics)
+WEIGHTS_PATH = "yolov8n.pt"  # Should be in your Space folder or auto-downloaded
 model = YOLO(WEIGHTS_PATH, task="detect")
-model.model = torch.load(WEIGHTS_PATH, map_location="cpu", weights_only=False)
 
 # Initialize EasyOCR
 reader = easyocr.Reader(['en'], gpu=False)
 
 def detect_and_read(image):
-    # YOLO detection
+    # Run YOLO detection
     results = model(image)
     detections = results[0].boxes.xyxy.cpu().numpy()  # x1, y1, x2, y2
     labels = results[0].names
@@ -45,7 +46,7 @@ demo = gr.Interface(
     fn=detect_and_read,
     inputs=gr.Image(type="numpy"),
     outputs="json",
-    title="YOLO + EasyOCR Demo",
+    title="YOLOv8 + EasyOCR",
     description="Object detection with YOLOv8 and OCR with EasyOCR"
 )