Update app.py

app.py

@@ -87,6 +87,8 @@ st.write(user_data_df)
 
 if st.button('Predict'):
     prediction = model.predict(user_data_df)
-    st.markdown('<div class="prediction">Placement Prediction: <span style="color: green;">Placed</span>' if prediction[0] == 1 else '<div class="prediction">Placement Prediction: <span style="color: red;">Not Placed</span>', unsafe_allow_html=True)
+    placement = "Placed" if prediction[0] == 1 else "Not Placed"
+    prediction_text = f"Placement Prediction: <span style='color: green;'>{placement}</span>" if placement == "Placed" else f"Placement Prediction: <span style='color: red;'>{placement}</span>"
+    st.markdown(f"<div class='prediction'>{prediction_text}</div>", unsafe_allow_html=True)
 
 st.markdown('</div>', unsafe_allow_html=True)