Update sentence_similarity_tab.py

tabs/sentence_similarity_tab.py

@@ -15,6 +15,7 @@ import networkx as nx
 from sklearn.manifold import TSNE
 from gensim.models import KeyedVectors
 from sentence_transformers import SentenceTransformer
+from sklearn.metrics.pairwise import cosine_similarity
 from translate_app import tr
 
 title = "Sentence Similarity"
@@ -379,5 +380,9 @@ def run():
     embeddings = model.encode(sentences)
     st.write(embeddings)
     st.write("")
+    # Calculate cosine similarity between the two sentences
+    similarity = cosine_similarity([embeddings[0]], [embeddings[1]])
+    st.write(f"Cosine similarity: {similarity[0][0]}")
+    st.write("")
     st.write("")
     st.write("")