UQ's picture
Update app.py
f417277 verified
import gradio as gr
import pickle
# Load the model and vectorizer
with open("model.pkl", "rb") as model_file:
model = pickle.load(model_file)
with open("vectorizer.pkl", "rb") as vectorizer_file:
vectorizer = pickle.load(vectorizer_file)
# Define the prediction function
def predict_sentiment(text):
text_vectorized = vectorizer.transform([text])
prediction = model.predict(text_vectorized)
sentiment = "Positive" if prediction == 1 else "Negative"
return sentiment
with gr.Blocks() as app:
gr.Markdown("<h1 style='text-align:center; color:black;'>Sentiment Analysis 💬</h1>")
gr.Markdown("<p style='text-align:center; color:black;'>Enter a sentence to check if it's positive or negative.</p>")
with gr.Row():
input_text = gr.Textbox(label="Your Text", placeholder="Type something here...", lines=2)
predict_button = gr.Button("Analyze Sentiment 🚀")
output_text = gr.Textbox(label="Prediction", interactive=False)
predict_button.click(fn=predict_sentiment, inputs=input_text, outputs=output_text)
app.launch(mcp_server=True,ssr_mode=False,share=True)