sberbank-ai commited on
Commit
448ab76
1 Parent(s): 8909d72

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -27,8 +27,8 @@ def mean_pooling(model_output, attention_mask):
27
  sentences = ['Привет! Как твои дела?',
28
  'А правда, что 42 твое любимое число?']
29
  #Load AutoModel from huggingface model repository
30
- tokenizer = AutoTokenizer.from_pretrained("sberbank-ai/ru_s_electra_medium")
31
- model = AutoModel.from_pretrained("sberbank-ai/ru_s_electra_medium")
32
  #Tokenize sentences
33
  encoded_input = tokenizer(sentences, padding=True, truncation=True, max_length=24, return_tensors='pt')
34
  #Compute token embeddings
 
27
  sentences = ['Привет! Как твои дела?',
28
  'А правда, что 42 твое любимое число?']
29
  #Load AutoModel from huggingface model repository
30
+ tokenizer = AutoTokenizer.from_pretrained("sberbank-ai/ruElectra-medium")
31
+ model = AutoModel.from_pretrained("sberbank-ai/ruElectra-medium")
32
  #Tokenize sentences
33
  encoded_input = tokenizer(sentences, padding=True, truncation=True, max_length=24, return_tensors='pt')
34
  #Compute token embeddings