ceejaytheanalyst commited on
Commit
f4165c6
·
verified ·
1 Parent(s): f1f67b1

Update app.py

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