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]
|