Update app.py

app.py

@@ -22,18 +22,14 @@ 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)) 
-
+def display_top_5(similarities):
     # 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")
@@ -43,11 +39,15 @@ word_list = st.text_area("Enter a list of words (one word per line):")
 
 if st.button("Analyze"):
     if reference_word and word_list:
-        words = word_list.splitlines()
-        for word in words:
+        # Calculate similarities for the reference phrase against the word list
+        similarities = []
+        for word in word_list.splitlines():
             similarity = calculate_similarity(reference_word, word)
-            st.write(f"Similarity between '{reference_word}' and '{word}': {similarity:.4f}")
-            display_top_5(reference_word, word_list)
+            similarities.append((word, similarity)) 
+
+        #  Find top 5 (We should only do this once outside the loop)
+        display_top_5(similarities) 
     else:
         st.warning("Please enter a reference word and a list of words.")
 
+