Update README.md
Browse files
README.md
CHANGED
@@ -49,7 +49,7 @@ To translate you can do:
|
|
49 |
|
50 |
```python
|
51 |
>>> sentence = "Entonces dijo"
|
52 |
-
>>> input = tokenizer(
|
53 |
>>> output = model.generate(input["input_ids"], max_length=40, num_beams=4, early_stopping=True)
|
54 |
>>> print('Original Sentence: {} \nTranslated sentence: {}'.format(sentence, tokenizer.decode(output[0])))
|
55 |
```
|
|
|
49 |
|
50 |
```python
|
51 |
>>> sentence = "Entonces dijo"
|
52 |
+
>>> input = tokenizer(sentence, return_tensors="pt")
|
53 |
>>> output = model.generate(input["input_ids"], max_length=40, num_beams=4, early_stopping=True)
|
54 |
>>> print('Original Sentence: {} \nTranslated sentence: {}'.format(sentence, tokenizer.decode(output[0])))
|
55 |
```
|