Image_Caption_Generator_flask_app

The following repository contains the code for a flask app that writes captions for Images. Two image to text models have been implemented BILP and ViT-GPT2.

Make sure you have the following libraries to run the code:

1. pytorch
2. transformers
3. Pillow
4. Flask
5. requests

To run the code, just run the app.py file.