Update README.md
Browse files
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).
|