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