Update app.py

app.py

@@ -70,6 +70,9 @@ def main():
         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)
+
         predicted_label = class_labels.get(predicted_class, "Unknown")
                 
         # Display the prediction result