| 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() | |