Ashishkr commited on
Commit
fba5599
1 Parent(s): e35fdfd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
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
- # Query sentences:
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]