Added class_name

app.py

@@ -18,8 +18,10 @@ def detect(image):
             x1, y1, x2, y2 = box.xyxy[0].tolist()  # Get box coordinates
             conf = float(box.conf[0])  # Get confidence
             cls = int(box.cls[0])  # Get class
+            class_name = model.names[cls]  # Get class name
             detections.append({
                 'class': cls,
+                'class_name': class_name,
                 'confidence': conf,
                 'box': [x1, y1, x2, y2]
             })
@@ -35,6 +37,6 @@ gr.Interface(
         gr.JSON(label="Detection Results")
     ],
     title="YOLOv9 (Ultralytics) Object Detection",
-    description="Upload an image to run object detection using your custom YOLOv9 model. The results will show both the annotated image and the detection details.",
+    description="Upload an image to run object detection using your custom YOLOv9 model. The results will show both the annotated image and the detection details including class names.",
 ).launch()