ceejaytheanalyst commited on
Commit
6d74074
·
verified ·
1 Parent(s): 18f7759

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,12 +15,12 @@ sentence_embed = pd.read_csv('Reference_file.csv')
15
  def mapping_code(user_input):
16
  emb1 = model.encode(user_input.lower())
17
  similarities = []
18
- for sentence_embed in sentences['embeds']:
19
- similarity = util.cos_sim(sentence_embed, emb1)
20
  similarities.append(similarity)
21
 
22
  # Combine similarity scores with 'code' and 'description'
23
- result = list(zip(sentences['SBS Code'], sentences['Long Description'], similarities))
24
 
25
  # Sort results by similarity scores
26
  result.sort(key=lambda x: x[2], reverse=True)
 
15
  def mapping_code(user_input):
16
  emb1 = model.encode(user_input.lower())
17
  similarities = []
18
+ for sentence in sentence_embed['embeds']:
19
+ similarity = util.cos_sim(sentence, emb1)
20
  similarities.append(similarity)
21
 
22
  # Combine similarity scores with 'code' and 'description'
23
+ result = list(zip(sentence_embed['SBS Code'], sentence_embed['Long Description'], similarities))
24
 
25
  # Sort results by similarity scores
26
  result.sort(key=lambda x: x[2], reverse=True)