ceejaytheanalyst commited on
Commit
f4090fc
·
verified ·
1 Parent(s): 1203285

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,8 @@ def mapping_code(user_input):
18
  emb1 = model.encode(user_input, convert_to_tensor=True)
19
  similarities = []
20
  for sentence in model.encode(sentence_embed['embeds'], convert_to_tensor=True):
21
- similarity = util.cos_sim(sentence, emb1)
 
22
  similarities.append(similarity)
23
 
24
  # Combine similarity scores with 'code' and 'description'
 
18
  emb1 = model.encode(user_input, convert_to_tensor=True)
19
  similarities = []
20
  for sentence in model.encode(sentence_embed['embeds'], convert_to_tensor=True):
21
+ #util.cos_sim(sentence, emb1)
22
+ similarity = util.paraphrase_mining(sentence, emb1,top_k=10)
23
  similarities.append(similarity)
24
 
25
  # Combine similarity scores with 'code' and 'description'