Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ def process_img(image):
|
|
29 |
result = model.predict(source=image)
|
30 |
|
31 |
if len(result[0].boxes)>0:
|
32 |
-
vis = plots.plot_detection(
|
33 |
class_names=class_names, min_score=0.2)
|
34 |
else:
|
35 |
vis = img
|
|
|
29 |
result = model.predict(source=image)
|
30 |
|
31 |
if len(result[0].boxes)>0:
|
32 |
+
vis = plots.plot_detection(image,boxes=result[0].boxes.boxes,
|
33 |
class_names=class_names, min_score=0.2)
|
34 |
else:
|
35 |
vis = img
|