Update app.py

app.py

@@ -43,7 +43,8 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # Load model
-pipe = pipeline("text-classification", model="NeonSamurai/hindi_bert_model")
+
+pipe = pipeline("text-classification", model="NeonSamurai/hindi_sentiment_bert_finetuned")
 
 names = ["neutral", "positive", "negative"]
 emojis = {"positive": "🤗", "negative": "😔", "neutral": "😐"}
@@ -90,9 +91,8 @@ if st.button("🔍 Predict Sentiment"):
         if not is_mostly_hindi(cleaned_input):
             st.error("❌ Input should be primarily in Hindi (Devanagari script).")
         else:
-            result = pipe(cleaned_input)[0]
-            sentiment_index = int(result['label'].split("_")[1])
-            sentiment = names[sentiment_index]
+            result = int(pipe(cleaned_input)[0]['label'].split("_")[1])
+            sentiment = names[result]
             emoji = emojis[sentiment]
 
             st.markdown(f"<div class='sentiment-result'>✅ Sentiment: <b>{sentiment.capitalize()} {emoji}</b></div>", unsafe_allow_html=True)