Update app.py

app.py

@@ -3,8 +3,8 @@ from ultralytics import YOLO
 import numpy as np
 import cv2
 
-# Load model
-model = YOLO("best.pt")   # Fire + Smoke YOLOv10 model
+# Load YOLOv10 Fire + Smoke model
+model = YOLO("best.pt") 
 
 def detect_fire_smoke(image):
     if image is None:
@@ -14,20 +14,21 @@ def detect_fire_smoke(image):
     results = model(img)[0]
 
     if len(results.boxes) == 0:
-        return "✔ SAFE — No Fire/Smoke Detected"
+        return "✔ SAFE — No Fire or Smoke Detected"
 
     output = []
+
     for box in results.boxes:
-        cls = int(box.cls[0])    # 0=fire, 1=smoke
+        cls_id = int(box.cls[0])  # YOLOv10 classes
         conf = float(box.conf[0])
 
-        if cls == 0:
+        if cls_id == 0:
             output.append(f"🔥 FIRE DETECTED — Confidence {conf:.2f}")
-        elif cls == 1:
+        elif cls_id == 1:
             output.append(f"💨 SMOKE DETECTED — Confidence {conf:.2f}")
 
     if not output:
-        return "✔ SAFE — No Fire/Smoke Detected"
+        return "✔ SAFE — No Fire or Smoke Detected"
 
     return "\n".join(output)
 
@@ -35,8 +36,8 @@ demo = gr.Interface(
     fn=detect_fire_smoke,
     inputs=gr.Image(type="pil"),
     outputs="text",
-    title="Fire & Smoke Detection",
-    description="Upload an image to detect fire or smoke using YOLOv10 model."
+    title="Fire & Smoke Detection (YOLOv10)",
+    description="Upload an image to detect fire or smoke."
 )
 
 demo.launch()