Nadun102's picture
Update app.py
6670fd8 verified
raw
history blame
382 Bytes
import gradio as gr
import joblib
model = joblib.load("model.joblib")
tfidf = joblib.load("tfidf_vectorizer.joblib")
def predict_sentiment(text):
vector = tfidf.transform([text])
prediction = model.predict(vector)[0]
return "Positive ✅" if prediction == 1 else "Negative ❌"
gr.Interface(
fn=predict_sentiment,
inputs="text",
outputs="text"
).launch()