dumperize
/

movie-picture-captioning

vision-encoder-decoder

Inference Endpoints

Model card Files Files and versions Community

dumperize commited on Jan 9, 2023

Commit

d6e6dba

•

1 Parent(s): d1be8cc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -31,7 +31,7 @@ We refined the model on the dataset with descriptions and movie posters by russi
 ```python
 from PIL import Image
 import torch
-from transformers import AutoTokenizer, AutoModel
 tokenizer = AutoTokenizer.from_pretrained("dumperize/movie-picture-captioning")
 feature_extractor = ViTFeatureExtractor.from_pretrained("dumperize/movie-picture-captioning")

 ```python
 from PIL import Image
 import torch
+from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
 tokenizer = AutoTokenizer.from_pretrained("dumperize/movie-picture-captioning")
 feature_extractor = ViTFeatureExtractor.from_pretrained("dumperize/movie-picture-captioning")