Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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"][
|
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 |
|