innocent-charles
commited on
Commit
•
0837bf6
1
Parent(s):
cb9297d
Update README.md
Browse files
README.md
CHANGED
@@ -151,9 +151,8 @@ print(embeddings)
|
|
151 |
import torch
|
152 |
from transformers import BertModel, BertTokenizerFast
|
153 |
|
154 |
-
|
155 |
-
|
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)
|