Update app.py

app.py

@@ -10,7 +10,7 @@ def download_models(model_id):
     return f"./{model_id}"
     
 box_annotator = sv.BoxAnnotator()
-category_dict = { }
+category_dict = {0: 'arborio', 1: 'ipsala', 2: 'jasmine', 3: 'karacadag', 4: 'basmati'}
 
 
 @spaces.GPU(duration=200)
@@ -21,7 +21,8 @@ def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold
     detections = sv.Detections.from_ultralytics(results)
     
     labels = [
-       'arborio', 'ipsala', 'jasmine', 'karacadag', 'basmati'
+        f"{category_dict[class_id]} {confidence:.2f}"
+        for class_id, confidence in zip(detections.class_id, detections.confidence)
     ]
     annotated_image = box_annotator.annotate(image, detections=detections, labels=labels)