demo-image-description-with-label

Sleeping

ydshieh HF Staff commited on Aug 21, 2023

Commit

4933944

1 Parent(s): 26cc724

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -201,7 +201,7 @@ def main():
         # By default, the generated  text is cleanup and the entities are extracted.
         processed_text, entities = processor.post_processor_generation(generated_text)
-        annotated_image = draw_entity_boxes_on_image(image_input, entities, show=True)
         color_id = -1
         entity_info = []

         # By default, the generated  text is cleanup and the entities are extracted.
         processed_text, entities = processor.post_processor_generation(generated_text)
+        annotated_image = draw_entity_boxes_on_image(image_input, entities, show=False)
         color_id = -1
         entity_info = []