tarekziade
commited on
Commit
•
e1927d3
1
Parent(s):
176eefb
Update README.md
Browse files
README.md
CHANGED
@@ -24,8 +24,8 @@ model-index:
|
|
24 |
type: image-to-text
|
25 |
name: Image To Text
|
26 |
dataset:
|
27 |
-
name:
|
28 |
-
type:
|
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
|
72 |
-
- DocOrNot
|
|
|
73 |
|
74 |
-
|
|
|
75 |
|
76 |
-
|
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
|
|
|
|
|
|
|
|
|
|