michelecafagna26
commited on
Commit
·
29cd1d4
1
Parent(s):
28fae9e
Update README.md
Browse files
README.md
CHANGED
@@ -13,7 +13,7 @@ metrics:
|
|
13 |
- rouge
|
14 |
library_name: transformers
|
15 |
---
|
16 |
-
## ClipCap fine-tuned for
|
17 |
|
18 |
[ClipCap](https://arxiv.org/abs/2111.09734) base trained on the [HL Dataset](https://huggingface.co/datasets/michelecafagna26/hl) for **high-level scene descriptions generation**
|
19 |
|
@@ -34,7 +34,7 @@ We fine-tune LM + Mapping Network starting from the model pretrained on COCO
|
|
34 |
|
35 |
## Demo
|
36 |
|
37 |
-
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/
|
38 |
|
39 |
## Installation
|
40 |
|
@@ -95,7 +95,7 @@ with torch.no_grad():
|
|
95 |
model.generate_beam(embed=prefix_embed)[0]
|
96 |
|
97 |
|
98 |
-
# >> ""
|
99 |
```
|
100 |
|
101 |
## BibTex and citation info
|
|
|
13 |
- rouge
|
14 |
library_name: transformers
|
15 |
---
|
16 |
+
## ClipCap fine-tuned for Scene Image Captioning
|
17 |
|
18 |
[ClipCap](https://arxiv.org/abs/2111.09734) base trained on the [HL Dataset](https://huggingface.co/datasets/michelecafagna26/hl) for **high-level scene descriptions generation**
|
19 |
|
|
|
34 |
|
35 |
## Demo
|
36 |
|
37 |
+
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1redkE4W2GusU_GVDuY_R2PXbh9ZDuTqf?usp=sharing)
|
38 |
|
39 |
## Installation
|
40 |
|
|
|
95 |
model.generate_beam(embed=prefix_embed)[0]
|
96 |
|
97 |
|
98 |
+
# >> "the picture is taken on the beach."
|
99 |
```
|
100 |
|
101 |
## BibTex and citation info
|