Update app.py

app.py

@@ -46,9 +46,11 @@ if st.button('Predict'):
     # Perform the prediction
     predicted_label, confidence = ensemble_predict(sentence)
 
-    # Display the result
-    st.write("Result:", predicted_label)
-    st.write("Confidence:", confidence)
-
+ # Display the result with custom styling
+    st.markdown(f"""
+    <h2 style='text-align: center; color: #1E90FF;'>Prediction Results</h2>
+    <p style='font-size: 24px; font-weight: bold;'>Result: <span style='color: #1E90FF;'>{predicted_label}</span></p>
+    <p style='font-size: 24px; font-weight: bold;'>Confidence: <span style='color: #1E90FF;'>{confidence:.2f}</span></p>
+    """, unsafe_allow_html=True)
 
 st.info("Remember: This prediction is not a diagnosis. Our method is designed to support, not replace, mental health professionals. The model's predictions should be used as a reference, and the final diagnosis should be made by a qualified professional to avoid potential biases and inaccuracies.")