innocent-charles commited on
Commit
0837bf6
1 Parent(s): cb9297d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -4
README.md CHANGED
@@ -151,9 +151,8 @@ print(embeddings)
151
  import torch
152
  from transformers import BertModel, BertTokenizerFast
153
 
154
-
155
- tokenizer = BertTokenizerFast.from_pretrained("setu4993/LaBSE")
156
- model = BertModel.from_pretrained("setu4993/LaBSE")
157
  model = model.eval()
158
 
159
  english_sentences = [
@@ -198,7 +197,6 @@ For similarity between sentences, an L2-norm is recommended before calculating t
198
  ```python
199
  import torch.nn.functional as F
200
 
201
-
202
  def similarity(embeddings_1, embeddings_2):
203
  normalized_embeddings_1 = F.normalize(embeddings_1, p=2)
204
  normalized_embeddings_2 = F.normalize(embeddings_2, p=2)
 
151
  import torch
152
  from transformers import BertModel, BertTokenizerFast
153
 
154
+ tokenizer = BertTokenizerFast.from_pretrained("sartifyllc/AviLaBSE")
155
+ model = BertModel.from_pretrained("sartifyllc/AviLaBSE")
 
156
  model = model.eval()
157
 
158
  english_sentences = [
 
197
  ```python
198
  import torch.nn.functional as F
199
 
 
200
  def similarity(embeddings_1, embeddings_2):
201
  normalized_embeddings_1 = F.normalize(embeddings_1, p=2)
202
  normalized_embeddings_2 = F.normalize(embeddings_2, p=2)