mp-02 commited on
Commit
5e2e7ba
·
verified ·
1 Parent(s): e1b2c07

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -75,4 +75,4 @@ def prediction(image):
75
  draw.rectangle(box)
76
  draw.text((box[0]+10, box[1]-10), text=prediction+ ", "+ str(confidence), font=font, fill="black", font_size="15")
77
 
78
- return d, image
 
75
  draw.rectangle(box)
76
  draw.text((box[0]+10, box[1]-10), text=prediction+ ", "+ str(confidence), font=font, fill="black", font_size="15")
77
 
78
+ return image, d