Image to Text

Generate a caption/description for your image, simple and straight forward using Transformers library.

How to use

• Create a Python 3.9 virtual environment
• Install all the packages from the requirements.txt file
• Download the model and supporting files from Huggingface
• Run the demo with gradio gradio_ui.py

HuggingFace Model

• https://huggingface.co/nlpconnect/vit-gpt2-image-captioning

Demo images

• https://unsplash.com/photos/46juD4zY1XA
• https://unsplash.com/photos/C8eSYwQkwHw
• https://unsplash.com/photos/PQewPJqNKwQ