hezarai
/

vit-gpt2-fa-image-captioning-flickr30k

Model card Files Files and versions Community

arxyzan commited on Sep 29, 2023

Commit

d480281

•

1 Parent(s): 8c868ea

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -8,7 +8,7 @@ pipeline_tag: image-to-text
 A Persian image captioning model constructed from a ViT + GPT2 architecture trained on flickr30k-fa.
 The encoder (ViT) was initialized from https://huggingface.co/google/vit-base-patch16-224 and the decoder (GPT2) was initialized
-from https://huggingface.co/HooshvareLab/roberta-fa-zwnj-base .
 ## Usage
 ```

 A Persian image captioning model constructed from a ViT + GPT2 architecture trained on flickr30k-fa.
 The encoder (ViT) was initialized from https://huggingface.co/google/vit-base-patch16-224 and the decoder (GPT2) was initialized
+from https://huggingface.co/HooshvareLab/gpt2-fa .
 ## Usage
 ```