| import gradio as gr | |
| from src import * | |
| model = ModelLoader() | |
| prediction = PredictionServices(model.Model, model.Tokenizer) | |
| def single_predict(text): | |
| print(text) | |
| preds = prediction.single_predict(text) | |
| toxic_pred = float(preds) | |
| non_toxic_pred = float(1-toxic_pred) | |
| rslt = {"Toxic":toxic_pred,"Non Toxic":non_toxic_pred} | |
| return rslt | |
| app = gr.Interface(inputs=gr.Textbox(label="Enter Comment"), | |
| fn=single_predict, | |
| outputs=[gr.Label('Probabilities')], | |
| title="Toxic Comment Classifier") | |
| app.launch() | |