Update README.md
Browse files
README.md
CHANGED
@@ -33,8 +33,8 @@ You can use this model with Transformers *pipeline*.
|
|
33 |
```python
|
34 |
from transformers import AutoTokenizer, pipeline
|
35 |
from optimum.onnxruntime import ORTModelForSeq2SeqLM
|
36 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
37 |
-
model = ORTModelForSeq2SeqLM.from_pretrained("
|
38 |
translator = pipeline("translation_en_to_fr", model=model, tokenizer=tokenizer)
|
39 |
results = translator("My name is Eustache and I have a pet raccoon")
|
40 |
print(results)
|
|
|
33 |
```python
|
34 |
from transformers import AutoTokenizer, pipeline
|
35 |
from optimum.onnxruntime import ORTModelForSeq2SeqLM
|
36 |
+
tokenizer = AutoTokenizer.from_pretrained("echarlaix/t5-small-onnx")
|
37 |
+
model = ORTModelForSeq2SeqLM.from_pretrained("echarlaix/t5-small-onnx")
|
38 |
translator = pipeline("translation_en_to_fr", model=model, tokenizer=tokenizer)
|
39 |
results = translator("My name is Eustache and I have a pet raccoon")
|
40 |
print(results)
|