Update README.md
Browse files
README.md
CHANGED
|
@@ -57,10 +57,9 @@ from sentence_transformers import SentenceTransformer, util
|
|
| 57 |
import torch
|
| 58 |
embedder = SentenceTransformer('paraphrase-distilroberta-base-v1')
|
| 59 |
|
| 60 |
-
# Corpus with example sentences
|
| 61 |
corpus = clarifications_gen
|
| 62 |
corpus_embeddings = embedder.encode(corpus, convert_to_tensor=True)
|
| 63 |
-
|
| 64 |
query = input_query.lower()
|
| 65 |
query_embedding = embedder.encode(query, convert_to_tensor=True)
|
| 66 |
cos_scores = util.pytorch_cos_sim(query_embedding, corpus_embeddings)[0]
|
|
|
|
| 57 |
import torch
|
| 58 |
embedder = SentenceTransformer('paraphrase-distilroberta-base-v1')
|
| 59 |
|
|
|
|
| 60 |
corpus = clarifications_gen
|
| 61 |
corpus_embeddings = embedder.encode(corpus, convert_to_tensor=True)
|
| 62 |
+
|
| 63 |
query = input_query.lower()
|
| 64 |
query_embedding = embedder.encode(query, convert_to_tensor=True)
|
| 65 |
cos_scores = util.pytorch_cos_sim(query_embedding, corpus_embeddings)[0]
|