Update app.py

app.py

@@ -19,14 +19,14 @@ def process_image(image):
         # Extract annotated image and labels with class names
         annotated_image = result.plot()
 
-        # Convert box.cls tensor to float before formatting
+        # Use cls attribute for labels and get class name from model, DO NOT use .item() on box.conf
         detected_areas_labels = "\n".join([
             f"{class_names[int(box.cls.item())].upper()}: {box.conf:.2f}" for box in result.boxes
         ])
 
         return annotated_image, detected_areas_labels
     except Exception as e:
-        return None, f"Error processing image: {e}"
+        return None, f"Error processing image: {e}
 
 
 # Create the Gradio Interface