Bruno Saraiva commited on
Commit
c3cd34c
1 Parent(s): 6fb7fa3

Add application file

Browse files
Files changed (2) hide show
  1. app.py +20 -0
  2. modelo_gradio.pk +0 -0
app.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pickle
2
+ import gradio
3
+
4
+ from sentence_transformers import SentenceTransformer
5
+ sentence_model = SentenceTransformer(
6
+ "sentence-transformers/paraphrase-multilingual-mpnet-base-v2")
7
+
8
+ file = open('modelo_gradio.pk', 'rb')
9
+ model_gradio = pickle.load(file)
10
+ file.close()
11
+
12
+ def is_racism(sentence):
13
+ s = sentence_model.encode(sentence, show_progress_bar=True)
14
+ pred = model_gradio.predict([s])
15
+ if pred == 1: return 'Dentro do Tema Racismo'
16
+ else: return 'Fora do Tema Racismo'
17
+
18
+ input = gradio.Textbox(label = "Input a Sentence")
19
+ app = gradio.Interface(fn = is_racism, inputs=[input], outputs="label")
20
+ app.launch()
modelo_gradio.pk ADDED
Binary file (507 kB). View file