Hugging Face's logo

hezarai
/
vit-roberta-fa-image-captioning-flickr30k

Image-to-Text
Hezar
Persian
Model card Files Community
1
arxyzan commited on
Commit
2a28b4d
•
1 Parent(s): e567714

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -8,6 +8,8 @@ tags:
8
  metrics:
9
  - wer
10
  pipeline_tag: image-to-text
 
 
11
  ---
12
 
13
  A Persian image captioning model constructed from a ViT + RoBERTa architecture trained on [flickr30k-fa](https://www.kaggle.com/datasets/sajjadayobi360/flickrfa) (created by Sajjad Ayoubi).
 
8
  metrics:
9
  - wer
10
  pipeline_tag: image-to-text
11
+ datasets:
12
+ - hezarai/flickr30k-fa
13
  ---
14
 
15
  A Persian image captioning model constructed from a ViT + RoBERTa architecture trained on [flickr30k-fa](https://www.kaggle.com/datasets/sajjadayobi360/flickrfa) (created by Sajjad Ayoubi).