ceejaytheanalyst commited on
Commit
e2a08b5
·
verified ·
1 Parent(s): 7fca223

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -30,7 +30,9 @@ def mapping_code(user_input):
30
  raise ValueError("Input sentence should be more than 1 word long.Please provide the Full description")
31
  emb1 = model.encode(user_input.lower())
32
  #similarities = util.pytorch_cos_sim(emb1, stored_embeddings)[0]
33
- similarity = util.cos_sim(sentence, emb1)
 
 
34
 
35
  # Combine similarity scores with 'code' and 'description'
36
  result = [(code, description, float(sim)) for code, description, sim in zip(stored_data["SBS_code"], stored_data["Description"], similarities)]
 
30
  raise ValueError("Input sentence should be more than 1 word long.Please provide the Full description")
31
  emb1 = model.encode(user_input.lower())
32
  #similarities = util.pytorch_cos_sim(emb1, stored_embeddings)[0]
33
+ for sentence in stored_embeddings:
34
+ similarity = util.cos_sim(sentence, emb1)
35
+ similarities.append(similarity)
36
 
37
  # Combine similarity scores with 'code' and 'description'
38
  result = [(code, description, float(sim)) for code, description, sim in zip(stored_data["SBS_code"], stored_data["Description"], similarities)]