rgera commited on
Commit
ff86e73
·
1 Parent(s): c94b000

small mistake

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(img, boxes=prediction["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)