Update app.py
Browse files
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 |
-
|
|
|
|
|
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)]
|