ai-forever commited on
Commit
3e5cc3d
1 Parent(s): 2e44b4c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -33,8 +33,8 @@ sentences = ['Привет! Как твои дела?',
33
  'А правда, что 42 твое любимое число?']
34
 
35
  #Load AutoModel from huggingface model repository
36
- tokenizer = AutoTokenizer.from_pretrained("sberbank-ai/sbert_large_nlu_ru")
37
- model = AutoModel.from_pretrained("sberbank-ai/sbert_large_nlu_ru")
38
 
39
  #Tokenize sentences
40
  encoded_input = tokenizer(sentences, padding=True, truncation=True, max_length=24, return_tensors='pt')
@@ -50,4 +50,4 @@ sentence_embeddings = mean_pooling(model_output, encoded_input['attention_mask']
50
  # Authors
51
  + [SberDevices](https://sberdevices.ru/) Team.
52
  + Denis Antykhov: [Github](https://github.com/gaphex);
53
- + Aleksandr Abramov: [Github](https://github.com/Ab1992ao), [Kaggle Competitions Master](https://www.kaggle.com/andrilko)
 
33
  'А правда, что 42 твое любимое число?']
34
 
35
  #Load AutoModel from huggingface model repository
36
+ tokenizer = AutoTokenizer.from_pretrained("ai-forever/sbert_large_nlu_ru")
37
+ model = AutoModel.from_pretrained("ai-forever/sbert_large_nlu_ru")
38
 
39
  #Tokenize sentences
40
  encoded_input = tokenizer(sentences, padding=True, truncation=True, max_length=24, return_tensors='pt')
 
50
  # Authors
51
  + [SberDevices](https://sberdevices.ru/) Team.
52
  + Denis Antykhov: [Github](https://github.com/gaphex);
53
+ + Aleksandr Abramov: [HF profile](https://huggingface.co/Andrilko), [Github](https://github.com/Ab1992ao), [Kaggle Competitions Master](https://www.kaggle.com/andrilko);