seanbenhur
commited on
Commit
•
51f72dc
1
Parent(s):
0a5d00b
fix the links
Browse files
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](https://huggingface.co/google/vit-base-patch16-224-in21k) encoder and [GPT2-Hindi](
|
4 |
|
5 |
This model was trained using HuggingFace course community week, organized by Huggingface.
|
6 |
|
@@ -51,4 +51,4 @@ This model was trained during HuggingFace course community week, organized by Hu
|
|
51 |
|
52 |
## Team Members
|
53 |
- [Sean Benhur](https://www.linkedin.com/in/seanbenhur/)
|
54 |
-
- [Herumb Shandilya](https://www.linkedin.com/in/
|
|
|
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 |
|
|
|
51 |
|
52 |
## Team Members
|
53 |
- [Sean Benhur](https://www.linkedin.com/in/seanbenhur/)
|
54 |
+
- [Herumb Shandilya](https://www.linkedin.com/in/herumb-s-740163131/)
|