Update app.py

app.py

@@ -69,14 +69,14 @@ def main():
         # Make prediction
         response = model.predict(processed_text)
         predicted_class = response.argmax(axis=-1)[0] 
-        # Map the predicted class index to the corresponding label
-        st.write("Predicted Index:", predicted_class)
-        st.write("Predicted Mental State:", predicted_label)
 
+        # Ensure the predicted_label is always assigned
         predicted_label = class_labels.get(predicted_class, "Unknown")
-                
-        # Display the prediction result
+
+        # Display the predicted index and label
+        st.write("Predicted Index:", predicted_class)
         st.write("Predicted Mental State:", predicted_label)
 
+
 if __name__ == "__main__":
     main()