echarlaix HF staff commited on
Commit
58de8de
1 Parent(s): d299f72

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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("optimum/t5-small")
37
- model = ORTModelForSeq2SeqLM.from_pretrained("optimum/t5-small")
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)