Update app.py

app.py

@@ -7,8 +7,8 @@ model = SentenceTransformer('sartifyllc/AviLaBSE')
 # Function to compute similarities
 def compute_similarity(original_sentence, sentences_to_compare):
     # Encode the original sentence and the sentences to compare
-    embeddings_original = model.encode([original_sentence])
-    embeddings_sentences_to_compare = model.encode(sentences_to_compare)
+    embeddings_original = model.encode([original_sentence], convert_to_tensor=True)
+    embeddings_sentences_to_compare = model.encode(sentences_to_compare, convert_to_tensor=True)
 
     # Compute cosine similarities
     similarities = util.cos_sim(embeddings_original, embeddings_sentences_to_compare)
@@ -34,7 +34,7 @@ iface = gr.Interface(
         gr.Textbox(label="Sentence 2"),
         gr.Textbox(label="Sentence 3")
     ],
-    outputs=gr.outputs.Table(label="Similarity Scores", columns=["Sentence", "Similarity Score"]),
+    outputs=gr.Table(label="Similarity Scores", headers=["Sentence", "Similarity Score"]),
     title="Sentence Similarity Calculator",
     description="Enter an original sentence and three sentences to compare their similarity."
 )