Update README.md
#1
by
ayushsinha
- opened
README.md
CHANGED
|
@@ -51,13 +51,13 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
|
| 51 |
|
| 52 |
|
| 53 |
# Inference Example
|
| 54 |
-
```
|
| 55 |
-
python
|
| 56 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 57 |
import torch
|
| 58 |
|
| 59 |
-
|
| 60 |
-
|
|
|
|
| 61 |
|
| 62 |
inputs = tokenizer("Hello, how are you?", return_tensors="pt").to("cuda")
|
| 63 |
outputs = model.generate(**inputs)
|
|
|
|
| 51 |
|
| 52 |
|
| 53 |
# Inference Example
|
| 54 |
+
```python
|
|
|
|
| 55 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 56 |
import torch
|
| 57 |
|
| 58 |
+
model_name = "AventIQ-AI/opus-mt-en-roa_multilanguageTranslation"
|
| 59 |
+
model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to("cuda")
|
| 60 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 61 |
|
| 62 |
inputs = tokenizer("Hello, how are you?", return_tensors="pt").to("cuda")
|
| 63 |
outputs = model.generate(**inputs)
|