lopesdri commited on
Commit
539c230
·
1 Parent(s): d74af8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def detect(img):
24
  boxes = predictions[:, :4] # x1, y1, x2, y2
25
  scores = predictions[:, 4]
26
  categories = predictions[:, 5]
27
- new_image = npnp.squeeze(results.render())
28
  print(new_image.shape)
29
  return new_image
30
 
 
24
  boxes = predictions[:, :4] # x1, y1, x2, y2
25
  scores = predictions[:, 4]
26
  categories = predictions[:, 5]
27
+ new_image = np.squeeze(results.render())
28
  print(new_image.shape)
29
  return new_image
30