Hugging Face's logo

team-indain-image-caption
/
hindi-image-captioning

Image-to-Text
Transformers
PyTorch
vision-encoder-decoder
Model card Files
xet
Community
2
seanbenhur commited on
Commit
0a5d00b
·
1 Parent(s): f8d3746

fix links

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -1,6 +1,6 @@
1
  # Hindi Image Captioning Model
2
 
3
- This is an encoder-decoder image captioning model made with VIT encoder and GPT2-Hindi as a decoder. This is a first attempt at using ViT + GPT2-Hindi for image captioning task. We used the Flickr8k Hindi Dataset available on kaggle to train the model.
4
 
5
  This model was trained using HuggingFace course community week, organized by Huggingface.
6
 
@@ -50,5 +50,5 @@ This model was trained during HuggingFace course community week, organized by Hu
50
  - Mixed Precision Enabled
51
 
52
  ## Team Members
53
- - [Sean Benhur]("https://www.linkedin.com/in/seanbenhur/")
54
- - [Herumb Shandilya]("https://www.linkedin.com/in/krypticmouse/")
 
1
  # Hindi Image Captioning Model
2
 
3
+ This is an encoder-decoder image captioning model made with [VIT](https://huggingface.co/google/vit-base-patch16-224-in21k) encoder and [GPT2-Hindi]("https://huggingface.co/surajp/gpt2-hindi) as a decoder. This is a first attempt at using ViT + GPT2-Hindi for image captioning task. We used the Flickr8k Hindi Dataset available on kaggle to train the model.
4
 
5
  This model was trained using HuggingFace course community week, organized by Huggingface.
6
 
 
50
  - Mixed Precision Enabled
51
 
52
  ## Team Members
53
+ - [Sean Benhur](https://www.linkedin.com/in/seanbenhur/)
54
+ - [Herumb Shandilya](https://www.linkedin.com/in/krypticmouse/)