cointegrated commited on
Commit
776805c
1 Parent(s): 2ce1adc

update the readme

Browse files
Files changed (1) hide show
  1. README.md +11 -0
README.md CHANGED
@@ -9,6 +9,8 @@ tags:
9
  - tiny
10
  - feature-extraction
11
  - sentence-similarity
 
 
12
 
13
  license: mit
14
  widget:
@@ -46,3 +48,12 @@ def embed_bert_cls(text, model, tokenizer):
46
  print(embed_bert_cls('привет мир', model, tokenizer).shape)
47
  # (312,)
48
  ```
 
 
 
 
 
 
 
 
 
9
  - tiny
10
  - feature-extraction
11
  - sentence-similarity
12
+ - sentence-transformers
13
+ - transformers
14
 
15
  license: mit
16
  widget:
48
  print(embed_bert_cls('привет мир', model, tokenizer).shape)
49
  # (312,)
50
  ```
51
+
52
+ Alternatively, you can use the model with `sentence_transformers`:
53
+ ```Python
54
+ from sentence_transformers import SentenceTransformer
55
+ model = SentenceTransformer('cointegrated/rubert-tiny2')
56
+ sentences = ["привет мир", "hello world", "здравствуй вселенная"]
57
+ embeddings = model.encode(sentences)
58
+ print(embeddings)
59
+ ```