Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| import joblib | |
| logistic_pipeline = joblib.load('logistic_pipeline.pkl') | |
| knn_pipeline = joblib.load('knn_pipeline.pkl') | |
| def predict_class_probabilities(text, model_choice): | |
| if model_choice == "Logistic Regression": | |
| probabilities = logistic_pipeline.predict_proba([text])[0] | |
| else: | |
| probabilities = knn_pipeline.predict_proba([text])[0] | |
| return {'Positive': probabilities[1], 'Negative': probabilities[0]} | |
| #Gradio interface | |
| input_text = gr.Textbox(lines=5, label="Enter your review comment") | |
| model_choice = gr.Dropdown(choices=["Logistic Regression", "KNN"], label="Choose a model") | |
| output_probs = gr.Label() | |
| iface = gr.Interface(fn=predict_class_probabilities, | |
| inputs=[input_text, model_choice], | |
| outputs=output_probs, | |
| title="Sentiment Emotion Predictor", | |
| description="Predict sentiment of review (positive or negative).") | |
| iface.launch() |