jcoelho commited on
Commit
d4e5f10
1 Parent(s): f19f58e

Create new file

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import sentence_transformers
3
+
4
+ model = sentence_transformers.SentenceTransformer('sentence-transformers/distiluse-base-multilingual-cased-v1')
5
+ def sim(s1, s2):
6
+ vec1 = model.encode(s1)
7
+ vec2 = model.encode(s2)
8
+ return sentence_transformers.util.pytorch_cos_sim(vec1, vec2 )[0][0].float()
9
+
10
+ iface = gr.Interface(sim, [gr.Textbox(label= "Frase 1", placeholder="Introduza a primeira frase..."), gr.Textbox(label= "Frase 2", placeholder="Introduza a segunda frase...")],
11
+ gr.Number(),
12
+ allow_flagging="auto",
13
+ title="Similaridade Semantica (PT)",
14
+ description="Dadas duas frases, retorna um valor entre 0 e 1, indicando 1 a similaridade máxima e 0 a mínima.",
15
+ examples=[["assalto", "ladrão"], ["assalto", "roubo"], ["assalto", "homicídio"], ["assalto", "computador"]])
16
+ iface.launch()