jcoelho's picture
Create new file
d4e5f10
raw
history blame contribute delete
No virus
919 Bytes
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()