File size: 587 Bytes
42053f5
 
 
 
 
 
 
 
 
 
 
 
 
 
e580c7a
 
42053f5
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import pickle
import gradio

from sentence_transformers import SentenceTransformer
sentence_model = SentenceTransformer(
    "sentence-transformers/paraphrase-multilingual-mpnet-base-v2")

file = open('modelo_gradio.pk', 'rb')
model_gradio = pickle.load(file)
file.close()

def is_racism(sentence):
	s = sentence_model.encode(sentence, show_progress_bar=True)
	pred = model_gradio.predict([s])
	if pred == 1: return 'Racist'
	else: return 'Unracist'

input = gradio.Textbox(label = "Input a Sentence")
app =  gradio.Interface(fn = is_racism, inputs=[input], outputs="label")
app.launch()