Update app.py

app.py

@@ -73,7 +73,7 @@ def main():
         predicted_label = class_labels.get(predicted_class, "Unknown")
                 
         # Display the prediction result
-        st.write("Predicted Mental State:", predicted_class)
+        st.write("Predicted Mental State:", predicted_label)
 
 if __name__ == "__main__":
     main()