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 'Racist' | |
| else: return 'Unracist' | |
| input = gradio.Textbox(label = "Input a Sentence") | |
| app = gradio.Interface(fn = is_racism, inputs=[input], outputs="label") | |
| app.launch() |