Update app/main.py

app/main.py

@@ -55,5 +55,5 @@ if st.button('Classify'):
     # Display the result
     prediction_raw_text.write("Prediction:")
     print(prediction)
-    prediction_text.write(f"The text is {label_map[prediction[0]['label']]} with a confidence of {prediction[0]['score']*100}%")
+    prediction_text.markdown(f"The text is <span style='color: {'red' if prediction[0]['label'] == 'LABEL_1' else 'green'}'>{label_map[prediction[0]['label']]}</span> with a confidence of {prediction[0]['score']*100}%", unsafe_allow_html=True)