Update app.py

app.py

@@ -7,6 +7,8 @@ from sklearn.manifold import TSNE
 import torch
 from transformers import AutoTokenizer, AutoModel
 import pandas as pd
+from sentence_transformers import SentenceTransformer, util  # Added util import
+
 
 # Page configuration
 st.set_page_config(layout="wide", page_title="Word & Sentence Embeddings Explorer")
@@ -162,9 +164,10 @@ def main():
                                       height=100)
             
             if compare_text:
+                # Calculate similarity using the imported util
                 similarity = util.pytorch_cos_sim(
-                    sent_model.encode(text_input),
-                    sent_model.encode(compare_text)
+                    sent_model.encode(text_input, convert_to_tensor=True),
+                    sent_model.encode(compare_text, convert_to_tensor=True)
                 ).item()
                 
                 st.metric(