| import sklearn | |
| import gradio as gr | |
| import joblib | |
| pipe = joblib.load("./pipeline.pkl") | |
| inputs = [gr.Textbox(value = "I love this!")] | |
| outputs = [gr.Label(label = "Sentiment")] | |
| title = "Sentiment Analysis Classifier" | |
| description = "This is a sentiment classifier using longformer model with a logistic regression head. " | |
| def infer(inputs): | |
| predictions = pipe.predict_proba([inputs]) | |
| label = { | |
| "negative":str(predictions[0][0]), | |
| "positive":str(predictions[0][1]), | |
| } | |
| return label | |
| gr.Interface(infer, inputs = inputs, outputs = outputs, title = title, description = description).launch() |