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()