ydshieh HF staff commited on
Commit
93e1f81
1 Parent(s): 9f18ecc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.post_processor_generation(generated_text)
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