tarekziade commited on
Commit
e1927d3
1 Parent(s): 176eefb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -11
README.md CHANGED
@@ -24,8 +24,8 @@ model-index:
24
  type: image-to-text
25
  name: Image To Text
26
  dataset:
27
- name: nlphuji/flickr30k
28
- type: nlphuji/flickr30k
29
  metrics:
30
  - name: ROUGE-1
31
  type: rouge
@@ -68,14 +68,11 @@ You can get that checkpoint using the 3083a3cef6e3c8dd90df3f088074bbe836b0f403 c
68
 
69
  It was then further fine-tuned on :
70
 
71
- - Flickr30k debiased: https://huggingface.co/datasets/Mozilla/flickr30k-transformed-captions
72
- - DocOrNot: https://huggingface.co/datasets/Mozilla/docornot
 
73
 
74
- You can find the code used to create the model here: https://github.com/mozilla/distilvit
 
75
 
76
- ### Framework versions
77
-
78
- - Transformers 4.40.2
79
- - Pytorch 2.3.0+cu121
80
- - Datasets 2.19.1
81
- - Tokenizers 0.19.1
 
24
  type: image-to-text
25
  name: Image To Text
26
  dataset:
27
+ name: Mozilla/flickr30k-transformed-captions
28
+ type: Mozilla/flickr30k-transformed-captions
29
  metrics:
30
  - name: ROUGE-1
31
  type: rouge
 
68
 
69
  It was then further fine-tuned on :
70
 
71
+ - [Flickr30k debiased](https://huggingface.co/datasets/Mozilla/flickr30k-transformed-captions)
72
+ - [DocOrNot](https://huggingface.co/datasets/Mozilla/docornot)
73
+ - [Alt Text Validation](https://huggingface.co/datasets/Mozilla/alt-text-validation)
74
 
75
+ For the latter, the dataset was annotated by our team to correct the alt text generated by the model,
76
+ using the [checkvite tool](https://github.com/mozila/checkvite).
77
 
78
+ You can find the code used to create the model here: https://github.com/mozilla/distilvit