team-indain-image-caption
/

hindi-image-captioning

Image-Text-to-Text

vision-encoder-decoder

Inference Endpoints

Model card Files Files and versions Community

seanbenhur commited on Nov 24, 2021

Commit

0a5d00b

•

1 Parent(s): f8d3746

fix links

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -1,6 +1,6 @@
 # Hindi Image Captioning Model
-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.
 This model was trained using HuggingFace course community week, organized by Huggingface.
@@ -50,5 +50,5 @@ This model was trained during HuggingFace course community week, organized by Hu
 - Mixed Precision Enabled
 ## Team Members
-- [Sean Benhur]("https://www.linkedin.com/in/seanbenhur/")
-- [Herumb Shandilya]("https://www.linkedin.com/in/krypticmouse/")

 # Hindi Image Captioning Model
+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.
 This model was trained using HuggingFace course community week, organized by Huggingface.
 - Mixed Precision Enabled
 ## Team Members
+- [Sean Benhur](https://www.linkedin.com/in/seanbenhur/)
+- [Herumb Shandilya](https://www.linkedin.com/in/krypticmouse/)