phamvi856 commited on
Commit
65341c5
1 Parent(s): fbd6b46

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ def process_image(image):
88
  predicted_label = iob_to_label(prediction)
89
  draw.rectangle(box, outline=label2color[predicted_label])
90
  draw.text((box[0]+10, box[1]-10), text=predicted_label, fill=label2color[predicted_label], font=font)
91
- draw.text((box[0]+10, box[1]+10), text=dataset[0]["ner_tags"][idx], fill='black', font=font)
92
 
93
  return image
94
 
 
88
  predicted_label = iob_to_label(prediction)
89
  draw.rectangle(box, outline=label2color[predicted_label])
90
  draw.text((box[0]+10, box[1]-10), text=predicted_label, fill=label2color[predicted_label], font=font)
91
+ draw.text((box[0]+10, box[1]+10), text=dataset[0]["ner_tags"][id], fill='black', font=font)
92
 
93
  return image
94