Update README.md
Browse files
README.md
CHANGED
|
@@ -29,6 +29,8 @@ We refined the model on the dataset with descriptions and movie posters by russi
|
|
| 29 |
# How to use
|
| 30 |
|
| 31 |
```python
|
|
|
|
|
|
|
| 32 |
from transformers import AutoTokenizer, AutoModel
|
| 33 |
|
| 34 |
tokenizer = AutoTokenizer.from_pretrained("dumperize/movie-picture-captioning")
|
|
|
|
| 29 |
# How to use
|
| 30 |
|
| 31 |
```python
|
| 32 |
+
from PIL import Image
|
| 33 |
+
import torch
|
| 34 |
from transformers import AutoTokenizer, AutoModel
|
| 35 |
|
| 36 |
tokenizer = AutoTokenizer.from_pretrained("dumperize/movie-picture-captioning")
|