README.md

Image Captioning

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

HuggingFace Model

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

execute the code "python gradio_ui.py"