Update README.md
Browse files
README.md
CHANGED
@@ -10,16 +10,16 @@ widget:
|
|
10 |
example_title: Custom Image Sample 1
|
11 |
---
|
12 |
# Model Card for movie-picture-captioning
|
13 |
-
The model generates descriptions of any photo in the style of movie descriptions.
|
14 |
|
15 |
# Model Details:
|
16 |
|
17 |
#### Model Description
|
18 |
|
19 |
-
This is an encoder
|
20 |
-
[Google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) was used as encoder, [DeepPavlov/rubert-base-cased](https://huggingface.co/DeepPavlov/rubert-base-cased) as decoder.
|
21 |
|
22 |
-
We refined the model on the dataset with descriptions and movie posters by russian app Kinoposk. Now the model generates descriptions
|
23 |
|
24 |
#### Model Sources
|
25 |
|
|
|
10 |
example_title: Custom Image Sample 1
|
11 |
---
|
12 |
# Model Card for movie-picture-captioning
|
13 |
+
The model generates descriptions of any photo in the style of movie descriptions. It learned from movie posters and descriptions. ... just for fun
|
14 |
|
15 |
# Model Details:
|
16 |
|
17 |
#### Model Description
|
18 |
|
19 |
+
This is an encoder-decoder model based on [VisionEncoderDecoderModel](https://huggingface.co/docs/transformers/model_doc/vision-encoder-decoder).
|
20 |
+
[Google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) was used as the encoder, [DeepPavlov/rubert-base-cased](https://huggingface.co/DeepPavlov/rubert-base-cased) as the decoder.
|
21 |
|
22 |
+
We refined the model on the dataset with descriptions and movie posters by the russian app Kinoposk. Now the model generates descriptions of the jargon of blockbusters =).
|
23 |
|
24 |
#### Model Sources
|
25 |
|