Update app.py

app.py

@@ -17,11 +17,8 @@ category_dict = {0: 'arborio', 1: 'basmati', 2: 'ipsala', 3: 'jasmine', 4: 'kara
 
 @spaces.GPU(duration=200)
 def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold):
-    rf = Roboflow(api_key="P03ZyTQgn7h7sH17vHoq")
-    project = rf.workspace().project("rice-detection-lgqqs")
-    model = project.version(2).model
-    # model_path = download_models(model_id)
-    # model = YOLOv10(model_path)
+    model_path = download_models(model_id)
+    model = YOLOv10(model_path)
     results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]
     detections = sv.Detections.from_ultralytics(results)