Spaces:
Sleeping
Sleeping
Update sroie_inference.py
Browse files- sroie_inference.py +1 -1
sroie_inference.py
CHANGED
@@ -96,7 +96,7 @@ def prediction(image):
|
|
96 |
#draw = ImageDraw.Draw(image, "RGBA")
|
97 |
font = ImageFont.load_default()
|
98 |
|
99 |
-
for prediction, box in zip(
|
100 |
draw.rectangle(box)
|
101 |
draw.text((box[0]+10, box[1]-10), text=prediction, font=font, fill="black", font_size="8")
|
102 |
|
|
|
96 |
#draw = ImageDraw.Draw(image, "RGBA")
|
97 |
font = ImageFont.load_default()
|
98 |
|
99 |
+
for prediction, box in zip(true_predictions, true_boxes):
|
100 |
draw.rectangle(box)
|
101 |
draw.text((box[0]+10, box[1]-10), text=prediction, font=font, fill="black", font_size="8")
|
102 |
|