import gradio as gr | |
from transformers.utils import logging | |
logging.set_verbosity_error() | |
from sentence_transformers import SentenceTransformer | |
from sentence_transformers import util | |
def compare(input): | |
embeddings1 = model.encode("Hello", convert_to_tensor=True) # TODO We need to find out how to split the input by line. | |
embeddings2 = model.encode(input, convert_to_tensor=True) | |
cosine_scores = util.cos_sim(embeddings1, embeddings2) | |
return "Score: {:.4f} \t\t {} \t\t {}".format(cosine_scores[0][0], sentences1[0], sentences2[0]) | |
model = SentenceTransformer("all-MiniLM-L6-v2") | |
demo = gr.Interface(fn=greet, inputs="text", outputs="text") | |
demo.launch() |