Update app.py

app.py

@@ -1,43 +1,30 @@
 import gradio as gr
-import torch
-from torch.serialization import add_safe_globals
 from ultralytics import YOLO
 import numpy as np
 import cv2
-from PIL import Image
-import ultralytics.nn.tasks as yolov8_tasks
 
-# ---------------------------
-# Allow YOLO model class for PyTorch 2.6+
-# ---------------------------
-add_safe_globals([yolov8_tasks.DetectionModel])
-
-# ---------------------------
-# Load model safely
-# ---------------------------
-model = YOLO("best.pt")   # Works after safe global patch
+# Load model
+model = YOLO("best.pt")   # Fire + Smoke YOLOv10 model
 
 def detect_fire_smoke(image):
     if image is None:
         return "Please upload an image"
 
     img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
-
     results = model(img)[0]
 
     if len(results.boxes) == 0:
         return "✔ SAFE — No Fire/Smoke Detected"
 
     output = []
-
     for box in results.boxes:
-        cls = int(box.cls[0])
+        cls = int(box.cls[0])    # 0=fire, 1=smoke
         conf = float(box.conf[0])
 
         if cls == 0:
-            output.append(f"🔥 FIRE DETECTED — Confidence: {conf:.2f}")
+            output.append(f"🔥 FIRE DETECTED — Confidence {conf:.2f}")
         elif cls == 1:
-            output.append(f"💨 SMOKE DETECTED — Confidence: {conf:.2f}")
+            output.append(f"💨 SMOKE DETECTED — Confidence {conf:.2f}")
 
     if not output:
         return "✔ SAFE — No Fire/Smoke Detected"