ai-forever
/

ruElectra-small

Inference Endpoints

Model card Files Files and versions Community

sberbank-ai commited on Feb 3, 2023

Commit

b24d25b

•

1 Parent(s): c9b97c0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ def mean_pooling(model_output, attention_mask):
 sentences = ['Привет! Как твои дела?',
              'А правда, что 42 твое любимое число?']
 #Load AutoModel from huggingface model repository
-tokenizer = AutoTokenizer.from_pretrained("sberbank-ai/ru_s_electra_small")
 model = AutoModel.from_pretrained("sberbank-ai/ruElectra-small")
 #Tokenize sentences
 encoded_input = tokenizer(sentences, padding=True, truncation=True, max_length=24, return_tensors='pt')

 sentences = ['Привет! Как твои дела?',
              'А правда, что 42 твое любимое число?']
 #Load AutoModel from huggingface model repository
+tokenizer = AutoTokenizer.from_pretrained("sberbank-ai/ruElectra-small")
 model = AutoModel.from_pretrained("sberbank-ai/ruElectra-small")
 #Tokenize sentences
 encoded_input = tokenizer(sentences, padding=True, truncation=True, max_length=24, return_tensors='pt')