Update app.py
Browse files
app.py
CHANGED
@@ -200,7 +200,7 @@ def main():
|
|
200 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
201 |
|
202 |
# By default, the generated text is cleanup and the entities are extracted.
|
203 |
-
processed_text, entities = processor.
|
204 |
|
205 |
annotated_image = draw_entity_boxes_on_image(image_input, entities, show=False)
|
206 |
|
|
|
200 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
201 |
|
202 |
# By default, the generated text is cleanup and the entities are extracted.
|
203 |
+
processed_text, entities = processor.post_process_generation(generated_text)
|
204 |
|
205 |
annotated_image = draw_entity_boxes_on_image(image_input, entities, show=False)
|
206 |
|