Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from transformers import pipeline
 
-# Attempt to load the model
+# Load emotion classification model
 @st.cache_resource
 def load_model():
     try:
@@ -101,9 +101,12 @@ def main():
     user_input = st.text_area("Enter a short sentence about your current mood:", "")
     
     if user_input:
+        # Clean the input text (stripping unnecessary spaces, lowercasing)
+        clean_input = user_input.strip().lower()
+        
         # Use the model to predict emotion
         try:
-            result = emotion_classifier(user_input)
+            result = emotion_classifier(clean_input)
             emotion = result[0]['label'].lower()
 
             st.subheader(f"Emotion Detected: {emotion.capitalize()}")