Update app.py

app.py

@@ -21,6 +21,20 @@ def calculate_similarity(word1, word2):
 
     cos_sim = torch.nn.functional.cosine_similarity(embeddings1, embeddings2, dim=0)
     return cos_sim.item()
+
+def display_top_5(reference_word, word_list):
+    similarities = []
+    for word in word_list.splitlines():
+        similarity = calculate_similarity(reference_word, word)
+        similarities.append((word, similarity)) 
+
+    # Sort by similarity (descending)
+    top_5_similarities = sorted(similarities, key=lambda item: item[1], reverse=True)[:5]
+
+    st.subheader("Top 5 Most Similar Words:")
+    for word, similarity in top_5_similarities:
+        st.write(f"- '{word}': {similarity:.4f}")
+        
 # Streamlit interface
 st.title("Word Similarity Checker")
 
@@ -33,5 +47,7 @@ if st.button("Analyze"):
         for word in words:
             similarity = calculate_similarity(reference_word, word)
             st.write(f"Similarity between '{reference_word}' and '{word}': {similarity:.4f}")
+            display_top_5(reference_word, word_list)
     else:
         st.warning("Please enter a reference word and a list of words.")
+