isaiahkabraham commited on
Commit
36cde7d
1 Parent(s): 9379e38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ logging.set_verbosity_error()
5
  from sentence_transformers import SentenceTransformer
6
 
7
  def compare(name):
8
- return cosine_scores[0][0]
9
 
10
  model = SentenceTransformer("all-MiniLM-L6-v2")
11
  sentences1 = ['The cat sits outside',
 
5
  from sentence_transformers import SentenceTransformer
6
 
7
  def compare(name):
8
+ return model.encode(name, convert_to_tensor=True)
9
 
10
  model = SentenceTransformer("all-MiniLM-L6-v2")
11
  sentences1 = ['The cat sits outside',