Update app.py

app.py

@@ -25,8 +25,8 @@ if user_input:
     logits = outputs.logits
     predicted_class_id = logits.argmax(-1).item()
 
-    # # Example sentiment mapping (you might have your own)
-    # sentiment_map = {0: "negative", 1: "neutral", 2: "positive"}
-    # sentiment = sentiment_map[predicted_class_id]
+    
+    sentiment_map = {0: "negative", 1: "neutral", 2: "positive"}  # Sentiment mapping
+    sentiment = sentiment_map[predicted_class_id]  # Get predicted sentiment 
 
-    st.write("outputs")
+    st.write(f"Predicted Sentiment: {sentiment}")  # Display the sentiment