Nadun102 commited on
Commit
6670fd8
·
verified ·
1 Parent(s): 4f73f1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -12
app.py CHANGED
@@ -1,22 +1,16 @@
1
  import gradio as gr
2
  import joblib
3
 
4
- # Load model & vectorizer
5
- model = joblib.load("sentiment_model.joblib")
6
  tfidf = joblib.load("tfidf_vectorizer.joblib")
7
 
8
  def predict_sentiment(text):
9
  vector = tfidf.transform([text])
10
  prediction = model.predict(vector)[0]
11
- label = " Positive" if prediction == 1 else "❌ Negative"
12
- return label
13
 
14
- iface = gr.Interface(
15
  fn=predict_sentiment,
16
- inputs=gr.Textbox(label="Enter a Review"),
17
- outputs=gr.Textbox(label="Sentiment"),
18
- title="XGBoost Sentiment Classifier",
19
- description="Real-time sentiment analysis using TF-IDF + XGBoost."
20
- )
21
-
22
- iface.launch()
 
1
  import gradio as gr
2
  import joblib
3
 
4
+ model = joblib.load("model.joblib")
 
5
  tfidf = joblib.load("tfidf_vectorizer.joblib")
6
 
7
  def predict_sentiment(text):
8
  vector = tfidf.transform([text])
9
  prediction = model.predict(vector)[0]
10
+ return "Positive ✅" if prediction == 1 else "Negative ❌"
 
11
 
12
+ gr.Interface(
13
  fn=predict_sentiment,
14
+ inputs="text",
15
+ outputs="text"
16
+ ).launch()