Spaces:
Running
Running
Fix supervision-related issue by downgrading to version 0.21.0 (#6)
Browse files- Fix supervision-related issue by downgrading to version 0.21.0 (54a6d3dd23e924a811ab14262a15722388cdb8ae)
Co-authored-by: Atalay Denknalbant <[email protected]>
- app.py +1 -1
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -25,7 +25,7 @@ category_dict = {
|
|
| 25 |
}
|
| 26 |
|
| 27 |
|
| 28 |
-
@spaces.GPU(
|
| 29 |
def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold):
|
| 30 |
model = YOLO(f"{model_id}.pt")
|
| 31 |
results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]
|
|
|
|
| 25 |
}
|
| 26 |
|
| 27 |
|
| 28 |
+
@spaces.GPU()
|
| 29 |
def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold):
|
| 30 |
model = YOLO(f"{model_id}.pt")
|
| 31 |
results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]
|
requirements.txt
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
supervision
|
| 2 |
ultralytics
|
| 3 |
spaces
|
| 4 |
gradio
|
|
|
|
| 1 |
+
supervision==0.21.0
|
| 2 |
ultralytics
|
| 3 |
spaces
|
| 4 |
gradio
|