Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
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 |
|