kedimestan
commited on
Commit
•
f043864
1
Parent(s):
7544046
Update app.py
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ def yolov8_inference(
|
|
54 |
current_model_name = model_name
|
55 |
model.overrides["conf"] = conf_threshold
|
56 |
model.overrides["iou"] = iou_threshold
|
57 |
-
results = model.predict(image, imgsz=image_size
|
58 |
|
59 |
boxes1 = []
|
60 |
for result in results:
|
|
|
54 |
current_model_name = model_name
|
55 |
model.overrides["conf"] = conf_threshold
|
56 |
model.overrides["iou"] = iou_threshold
|
57 |
+
results = model.predict(image, imgsz=image_size)
|
58 |
|
59 |
boxes1 = []
|
60 |
for result in results:
|