Update app.py

app.py

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