/Image-Description-Generation

Use of nlpconnect/vit-gpt2-image-captioning pre-trained model from huggingface to generate descriptions for images.

Primary LanguageJupyter Notebook

Watchers