supermy commited on
Commit
ef6903d
1 Parent(s): 7d8f9b4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -38,9 +38,9 @@ widget:
38
  Here is how to use this model to get the features of a given text in PyTorch:
39
 
40
  ```python
41
- from transformers import AutoTokenizer, AutoModelForCausalLM
42
  tokenizer = AutoTokenizer.from_pretrained("supermy/couplet-helsinki")
43
- model = AutoModelForCausalLM.from_pretrained("supermy/couplet-helsinki")
44
  ```
45
 
46
 
 
38
  Here is how to use this model to get the features of a given text in PyTorch:
39
 
40
  ```python
41
+ from transformers import AutoTokenizer, AutoModel
42
  tokenizer = AutoTokenizer.from_pretrained("supermy/couplet-helsinki")
43
+ model = AutoModel.from_pretrained("supermy/couplet-helsinki")
44
  ```
45
 
46