Update app.py
Browse files
app.py
CHANGED
@@ -20,10 +20,10 @@ def cosine_similarity(v1, v2):
|
|
20 |
|
21 |
# Backend function for mapping
|
22 |
def mapping_code(user_input):
|
23 |
-
emb1 = model.encode(user_input, convert_to_tensor=True)
|
24 |
similarities = []
|
25 |
for sentence_emb in sentence_embed['embeds']:
|
26 |
-
sentence_emb = np.array(sentence_emb)
|
27 |
similarity = cosine_similarity(sentence_emb, emb1)
|
28 |
similarities.append(similarity)
|
29 |
|
@@ -40,6 +40,7 @@ def mapping_code(user_input):
|
|
40 |
top_5_results.append({"Code": code, "Description": description, "Similarity Score": similarity_score})
|
41 |
return top_5_results
|
42 |
|
|
|
43 |
# Streamlit frontend interface
|
44 |
def main():
|
45 |
st.title("CPT Description Mapping")
|
|
|
20 |
|
21 |
# Backend function for mapping
|
22 |
def mapping_code(user_input):
|
23 |
+
emb1 = model.encode(user_input, convert_to_tensor=True).astype(float)
|
24 |
similarities = []
|
25 |
for sentence_emb in sentence_embed['embeds']:
|
26 |
+
sentence_emb = np.array(sentence_emb).astype(float)
|
27 |
similarity = cosine_similarity(sentence_emb, emb1)
|
28 |
similarities.append(similarity)
|
29 |
|
|
|
40 |
top_5_results.append({"Code": code, "Description": description, "Similarity Score": similarity_score})
|
41 |
return top_5_results
|
42 |
|
43 |
+
|
44 |
# Streamlit frontend interface
|
45 |
def main():
|
46 |
st.title("CPT Description Mapping")
|