Kosmos-2 not extracting entites???

NielsRogge / Transformers-Tutorials

This repository contains demos I made with the Transformers library by HuggingFace.

MIT License

9.16k stars 1.42k forks source link

Kosmos-2 not extracting entites??? #378

Open andysingal opened 8 months ago

andysingal commented 8 months ago

While running

# By default, the generated text is cleaned up and the entities are extracted.
processed_text, entities = processor.post_process_generation(generated_text)

print(processed_text)
print(entities)

i got

An image of a snowman warming up by a fire.
[]