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