Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -60,7 +60,7 @@ def yolo(size, iou, conf, im):
|
|
| 60 |
results2 = model(im) # inference
|
| 61 |
|
| 62 |
results2.render() # updates results.imgs with boxes and labels
|
| 63 |
-
return Image.fromarray(results2.
|
| 64 |
|
| 65 |
#------------ Interface-------------
|
| 66 |
|
|
|
|
| 60 |
results2 = model(im) # inference
|
| 61 |
|
| 62 |
results2.render() # updates results.imgs with boxes and labels
|
| 63 |
+
return Image.fromarray(results2.ims[0])
|
| 64 |
|
| 65 |
#------------ Interface-------------
|
| 66 |
|