import gradio as gr
from sentence_transformers import SentenceTransformer, util

model = SentenceTransformer('sentence-transformers/paraphrase-mpnet-base-v2')

def compute_similarity(text1, text2):
    embeddings = model.encode([text1, text2])
    similarity = util.cos_sim(embeddings[0], embeddings[1])
    return float(similarity)

iface = gr.Interface(
    fn=compute_similarity,
    inputs=["text", "text"],
    outputs="number",
    title="Sentence Similarity with paraphrase-mpnet-base-v2",
    description="Compute the cosine similarity between two sentences."
)

iface.launch()