paugar commited on
Commit
7024583
·
verified ·
1 Parent(s): 0d3a1e8

Update app.py

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