hezarai
/

vit-roberta-fa-image-captioning-flickr30k

Model card Files Files and versions Community

arxyzan commited on Sep 29, 2023

Commit

039ce81

•

1 Parent(s): a51bbf9

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -17,7 +17,7 @@ pip install hezar
 ```python
 from hezar import Model
-model = Model.load("hezarai/vit-gpt2-fa-image-captioning-flickr30k")
 captions = model.predict("example_image.jpg")
 print(captions)
 ```

 ```python
 from hezar import Model
+model = Model.load("hezarai/vit-roberta-fa-image-captioning-flickr30k")
 captions = model.predict("example_image.jpg")
 print(captions)
 ```