AdrielAmoguis commited on
Commit
7487879
·
1 Parent(s): eab0e5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def snap(image, model, conf, iou):
25
  result = results[0]
26
 
27
  if result.boxes.cls.cpu().numpy().size == 0:
28
- return [Image.fromarray(image)]
29
  classes = result.boxes.cls.cpu().numpy()[0]
30
  probs = result.boxes.conf.cpu().numpy()[0]
31
  boxes = result.boxes.xyxy.cpu().numpy()
 
25
  result = results[0]
26
 
27
  if result.boxes.cls.cpu().numpy().size == 0:
28
+ return [image]
29
  classes = result.boxes.cls.cpu().numpy()[0]
30
  probs = result.boxes.conf.cpu().numpy()[0]
31
  boxes = result.boxes.xyxy.cpu().numpy()