hezarai
/

vit-gpt2-fa-image-captioning-flickr30k

Model card Files Files and versions Community

arxyzan commited on Dec 10, 2023

Commit

bf0b1a7

•

1 Parent(s): 321898c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

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

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