Spaces:
Runtime error
Runtime error
small mistake
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ def predict(inp1):
|
|
| 26 |
# Step 4: Use the model and visualize the prediction
|
| 27 |
prediction = model(batch)[0]
|
| 28 |
labels = [weights.meta["categories"][i] for i in prediction["labels"]]
|
| 29 |
-
box = draw_bounding_boxes(
|
| 30 |
labels=labels,
|
| 31 |
colors="red",
|
| 32 |
width=4, font_size=30)
|
|
|
|
| 26 |
# Step 4: Use the model and visualize the prediction
|
| 27 |
prediction = model(batch)[0]
|
| 28 |
labels = [weights.meta["categories"][i] for i in prediction["labels"]]
|
| 29 |
+
box = draw_bounding_boxes(inp1, boxes=prediction["boxes"],
|
| 30 |
labels=labels,
|
| 31 |
colors="red",
|
| 32 |
width=4, font_size=30)
|