Spaces:
Runtime error
Runtime error
File size: 615 Bytes
c3cd34c |
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 '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() |