Similarity / app.py
jeffrey1963's picture
Update app.py
ba04b68 verified
raw
history blame contribute delete
764 Bytes
import gradio as gr
from sentence_transformers import SentenceTransformer, util
# Use a more powerful model for paraphrasing and semantic similarity
model = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")
def compare_sentences(sentence_a, sentence_b):
emb = model.encode([sentence_a, sentence_b], convert_to_tensor=True)
score = util.pytorch_cos_sim(emb[0], emb[1]).item()
return f"{score * 100:.2f}%" # Similarity score as percentage
iface = gr.Interface(
fn=compare_sentences,
inputs=["text", "text"],
outputs="text",
title="Sentence Similarity Checker (MPNet Model)",
description="Enter two sentences to compare their meaning using a powerful transformer model. Score is shown as a percent."
)
iface.launch()