|
import gradio as gr |
|
from sentence_transformers import SentenceTransformer, util |
|
model_id = "sentence-transformers/multi-qa-mpnet-base-dot-v1" |
|
|
|
model = SentenceTransformer( |
|
model_id |
|
) |
|
|
|
def launch(source_sentence, sentences): |
|
source = model.encode(source_sentence, convert_to_tensor=True) |
|
references = model.encode([e.strip() for e in sentences.split("|")], convert_to_tensor=True) |
|
return ",".join([str(e) for e in util.pytorch_cos_sim(source, references).flatten().tolist()]) |
|
|
|
iface = gr.Interface(launch, inputs=["text","text"], outputs="text") |
|
iface.launch() |