Update README.md
Browse files
README.md
CHANGED
@@ -5,10 +5,10 @@ language:
|
|
5 |
pipeline_tag: image-to-text
|
6 |
widget:
|
7 |
- src: >-
|
8 |
-
https://huggingface.co/snzhang/
|
9 |
example_title: SpiderMan
|
10 |
- src: >-
|
11 |
-
https://huggingface.co/snzhang/
|
12 |
example_title: Born to Fly
|
13 |
---
|
14 |
|
@@ -28,7 +28,7 @@ The training data contains 5043 movie posts and their corresponding Chinese titl
|
|
28 |
from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer
|
29 |
from PIL import Image
|
30 |
|
31 |
-
pretrained = "snzhang/
|
32 |
model = VisionEncoderDecoderModel.from_pretrained(pretrained)
|
33 |
feature_extractor = ViTFeatureExtractor.from_pretrained(pretrained)
|
34 |
tokenizer = AutoTokenizer.from_pretrained(pretrained)
|
@@ -47,4 +47,4 @@ print(preds)
|
|
47 |
|
48 |
## More Details
|
49 |
|
50 |
-
You can get more training details in [
|
|
|
5 |
pipeline_tag: image-to-text
|
6 |
widget:
|
7 |
- src: >-
|
8 |
+
https://huggingface.co/snzhang/FilmTitle-Beit-GPT2/resolve/main/SpiderMan.jpg
|
9 |
example_title: SpiderMan
|
10 |
- src: >-
|
11 |
+
https://huggingface.co/snzhang/FilmTitle-Beit-GPT2/resolve/main/BorntoFly.jpg
|
12 |
example_title: Born to Fly
|
13 |
---
|
14 |
|
|
|
28 |
from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer
|
29 |
from PIL import Image
|
30 |
|
31 |
+
pretrained = "snzhang/FilmTitle-Beit-GPT2"
|
32 |
model = VisionEncoderDecoderModel.from_pretrained(pretrained)
|
33 |
feature_extractor = ViTFeatureExtractor.from_pretrained(pretrained)
|
34 |
tokenizer = AutoTokenizer.from_pretrained(pretrained)
|
|
|
47 |
|
48 |
## More Details
|
49 |
|
50 |
+
You can get more training details in [FilmTitle-Beit-GPT2](https://github.com/h7nian/FilmTitle-Beit-GPT2)
|