|
--- |
|
language: |
|
- fa |
|
metrics: |
|
- wer |
|
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/gpt2-fa . |
|
|
|
## Usage |
|
``` |
|
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) |
|
``` |
