arxyzan commited on
Commit
321898c
1 Parent(s): ad4b94b

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 + GPT2 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 + GPT2 architecture trained on [flickr30k-fa](https://www.kaggle.com/datasets/sajjadayobi360/flickrfa) (created by Sajjad Ayoubi).