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