Commit
·
7487879
1
Parent(s):
eab0e5c
Update app.py
Browse files
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 [
|
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()
|