Update app.py

app.py

@@ -91,9 +91,9 @@ def image_annotate(image:str, annotator:str) -> Image.Image:
 
   elif annotator == "Label":
     labels = [
-    f"{class_name} {confidence:.2f}"
-    for class_name, confidence
-    in zip(detections.class_name, detections.confidence)
+    f"{class_id} {confidence:.2f}"
+    for class_id, confidence
+    in zip(detections.class_id, detections.confidence)
     ]
 
     rich_label_annotator = sv.RichLabelAnnotator(