rasoulasadianub commited on
Commit
830e687
·
verified ·
1 Parent(s): dcaf21a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
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 = "shenasa/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,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)