Spaces:
Runtime error
Runtime error
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 'Dentro do Tema Racismo' | |
else: return 'Fora do Tema Racismo' | |
input = gradio.Textbox(label = "Input a Sentence") | |
app = gradio.Interface(fn = is_racism, inputs=[input], outputs="label") | |
app.launch() |