Update README.md
Browse files
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)
|