Fix a code issue
#4
by
Yehor
- opened
README.md
CHANGED
@@ -64,6 +64,6 @@ model = AutoModelForSeq2SeqLM.from_pretrained("grammarly/spivavtor-large")
|
|
64 |
input_text = 'Перефразуйте речення: Який найкращий комплімент, який ти отримував від будь-кого?'
|
65 |
|
66 |
inputs = tokenizer.encode(input_text, return_tensors="pt")
|
67 |
-
|
68 |
output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
69 |
```
|
|
|
64 |
input_text = 'Перефразуйте речення: Який найкращий комплімент, який ти отримував від будь-кого?'
|
65 |
|
66 |
inputs = tokenizer.encode(input_text, return_tensors="pt")
|
67 |
+
outputs = model.generate(inputs, max_length=256)
|
68 |
output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
69 |
```
|