Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| def score_comment(comment): | |
| vectorized_comment = vectorizer([comment]) | |
| results = model.predict(vectorized_comment) | |
| text = '' | |
| for idx, col in enumerate(df.columns[2:]): | |
| text += '{}: {}\n'.format(col, results[0][idx]>0.5) | |
| return text | |
| interface = gr.Interface(fn = score_comment, inputs=gr.Textbox(lines=2, placeholder='Comment to score'), outputs = 'text', title='Comment Toxicity', description='Check comment, how much is it toxic:') | |
| interface.launch(share=True) |