Update README.md
Browse files
README.md
CHANGED
|
@@ -45,7 +45,7 @@ import arabic_reshaper
|
|
| 45 |
from bidi.algorithm import get_display
|
| 46 |
import matplotlib.pyplot as plt
|
| 47 |
|
| 48 |
-
model_name = "
|
| 49 |
model = VisionEncoderDecoderModel.from_pretrained(model_name)
|
| 50 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 51 |
tokenizer.pad_token_id = tokenizer.eos_token_id
|
|
@@ -99,5 +99,4 @@ If you use this model, please cite the original paper:
|
|
| 99 |
- **Repository**: [GitHub - PTIR](https://github.com/rasoulasadiyan/PTIR)
|
| 100 |
- **Demo**: Available at [PTIR Demo](https://rasoulasadiyan.github.io/PTIR)
|
| 101 |
- **Related Work**: Based on prior implementations like [PIC in TensorFlow](https://github.com/rasoulasadiyan/Persian-Image-Captioning-PIC)
|
| 102 |
-
- **Dataset**: [COCO-PIC Dataset](https://huggingface.co/datasets/rasoulasadianub/coco-pic)
|
| 103 |
-
- **Acknowledgments**: This work advances Persian AI resources, building on open-source tools like Hugging Face and Milvus.
|
|
|
|
| 45 |
from bidi.algorithm import get_display
|
| 46 |
import matplotlib.pyplot as plt
|
| 47 |
|
| 48 |
+
model_name = "rasoulasadianub/Persian-Image-Captioning"
|
| 49 |
model = VisionEncoderDecoderModel.from_pretrained(model_name)
|
| 50 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 51 |
tokenizer.pad_token_id = tokenizer.eos_token_id
|
|
|
|
| 99 |
- **Repository**: [GitHub - PTIR](https://github.com/rasoulasadiyan/PTIR)
|
| 100 |
- **Demo**: Available at [PTIR Demo](https://rasoulasadiyan.github.io/PTIR)
|
| 101 |
- **Related Work**: Based on prior implementations like [PIC in TensorFlow](https://github.com/rasoulasadiyan/Persian-Image-Captioning-PIC)
|
| 102 |
+
- **Dataset**: [COCO-PIC Dataset](https://huggingface.co/datasets/rasoulasadianub/coco-pic)
|
|
|