Update app.py

app.py

@@ -0,0 +1,33 @@
+import gradio as gr
+from sentence_transformers import SentenceTransformer, util
+
+# Load the model
+model = SentenceTransformer('sartifyllc/AviLaBSE')
+
+# Function to compute similarities
+def compute_similarities(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)
+
+    # Compute cosine similarities
+    similarities = util.cos_sim(embeddings_original, embeddings_sentences_to_compare)
+
+    # Prepare the results as a list of tuples
+    results = [(sentence, similarities[0][i].item()) for i, sentence in enumerate(sentences_to_compare)]
+    return results
+
+# Define the Gradio interface
+iface = gr.Interface(
+    fn=compute_similarities,
+    inputs=[
+        gr.inputs.Textbox(lines=2, placeholder="Enter the original sentence here..."),
+        gr.inputs.Textbox(lines=5, placeholder="Enter sentences to compare, separated by new lines...")
+    ],
+    outputs=gr.outputs.Dataframe(headers=["Sentence", "Similarity Score"]),
+    title="Sentence Similarity Checker",
+    description="Enter an original sentence and a list of sentences to compare. The app will compute and display similarity scores for each comparison."
+)
+
+# Launch the interface
+iface.launch()