Artwork Caption Generator

• Tech Stack:
• Language: Python
• Libraries: Pytorch, Transformers, HuggingFace, Selenium, Pandas, FastAI
• IDE: VS Code
• Notebook: Google Colab
• Deployment: Link
• Github URL: Project Link

By collecting artworks and their relevant title given by the artists and training a generative AI, I created an artwork caption generator that can generate caption for the created art. I used Swin Transformer model as the encoder and GPT-2 as the decoder model.