Spaces:
Runtime error
Runtime error
import gradio as gr | |
import sentence_transformers | |
model = sentence_transformers.SentenceTransformer('sentence-transformers/distiluse-base-multilingual-cased-v1') | |
def sim(s1, s2): | |
vec1 = model.encode(s1) | |
vec2 = model.encode(s2) | |
return sentence_transformers.util.pytorch_cos_sim(vec1, vec2 )[0][0].float() | |
iface = gr.Interface(sim, [gr.Textbox(label= "Frase 1", placeholder="Introduza a primeira frase..."), gr.Textbox(label= "Frase 2", placeholder="Introduza a segunda frase...")], | |
gr.Number(), | |
allow_flagging="auto", | |
title="Similaridade Semantica (PT)", | |
description="Dadas duas frases, retorna um valor entre 0 e 1, indicando 1 a similaridade máxima e 0 a mínima.", | |
examples=[["assalto", "ladrão"], ["assalto", "roubo"], ["assalto", "homicídio"], ["assalto", "computador"]]) | |
iface.launch() |