BubbleSheep commited on
Commit
f1793d1
1 Parent(s): ebc5c9c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -25,9 +25,11 @@ This model has been pre-trained for English-Chinese Translation, and use dataset
25
  ## How to Get Started With the Model
26
 
27
  ```python
28
- from transformers import AutoTokenizer, AutoModelForMaskedLM
 
29
  tokenizer = AutoTokenizer.from_pretrained("BubbleSheep/Hgn_trans_en2zh")
30
- model = AutoModelForMaskedLM.from_pretrained("BubbleSheep/Hgn_trans_en2zh")
 
31
 
32
  ```
33
 
 
25
  ## How to Get Started With the Model
26
 
27
  ```python
28
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
29
+
30
  tokenizer = AutoTokenizer.from_pretrained("BubbleSheep/Hgn_trans_en2zh")
31
+
32
+ model = AutoModelForSeq2SeqLM.from_pretrained("BubbleSheep/Hgn_trans_en2zh")
33
 
34
  ```
35