paugar commited on
Commit
7147635
·
verified ·
1 Parent(s): d98a894

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.boxes)>0:
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: