Update README.md
Browse files
README.md
CHANGED
|
@@ -1,7 +1,9 @@
|
|
| 1 |
---
|
| 2 |
-
license: cc-by-3.0
|
| 3 |
language:
|
| 4 |
- en
|
|
|
|
|
|
|
|
|
|
| 5 |
---
|
| 6 |
|
| 7 |
A model for mapping abstract sentence descriptions to sentences that fit the descriptions. Trained on Wikipedia. Use ```load_finetuned_model``` to load the query and sentence encoder, and ```encode_batch()``` to encode a sentence with the model.
|
|
@@ -14,9 +16,9 @@ import torch
|
|
| 14 |
def load_finetuned_model():
|
| 15 |
|
| 16 |
|
| 17 |
-
sentence_encoder = AutoModel.from_pretrained("
|
| 18 |
-
query_encoder = AutoModel.from_pretrained("
|
| 19 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 20 |
|
| 21 |
return tokenizer, query_encoder, sentence_encoder
|
| 22 |
|
|
|
|
| 1 |
---
|
|
|
|
| 2 |
language:
|
| 3 |
- en
|
| 4 |
+
tags:
|
| 5 |
+
- Sentence Similarity
|
| 6 |
+
- feature-extraction
|
| 7 |
---
|
| 8 |
|
| 9 |
A model for mapping abstract sentence descriptions to sentences that fit the descriptions. Trained on Wikipedia. Use ```load_finetuned_model``` to load the query and sentence encoder, and ```encode_batch()``` to encode a sentence with the model.
|
|
|
|
| 16 |
def load_finetuned_model():
|
| 17 |
|
| 18 |
|
| 19 |
+
sentence_encoder = AutoModel.from_pretrained("biu-nlp/abstract-sim-sentence")
|
| 20 |
+
query_encoder = AutoModel.from_pretrained("biu-nlp/abstract-sim-query")
|
| 21 |
+
tokenizer = AutoTokenizer.from_pretrained("biu-nlp/abstract-sim-sentence")
|
| 22 |
|
| 23 |
return tokenizer, query_encoder, sentence_encoder
|
| 24 |
|