Update cancer.py

cancer.py

@@ -97,10 +97,16 @@ if st.button("Predict Cancer Presence"):
     if 'trained_model' in st.session_state:
         model = st.session_state['trained_model']
         x_train = st.session_state['x_train']
+
+        # Create DataFrame for input
         input_df = pd.DataFrame(input_data, columns=x_train.columns)
+
+        # Apply preprocessing (fix error)
         input_transformed = model.named_steps['preprocessor'].transform(input_df)
+
+        # Make prediction
         prediction = model.named_steps['classifier'].predict(input_transformed)
-        
+
         if prediction[0] == 1:
             st.markdown("<h3 style='color: red;'>Cancer Prediction: Positive 🟥</h3>", unsafe_allow_html=True)
             st.write("Unfortunately, the model predicts the presence of cancer. Please consult a doctor for further advice.")
@@ -108,4 +114,4 @@ if st.button("Predict Cancer Presence"):
             st.markdown("<h3 style='color: green;'>Cancer Prediction: Negative 🟩</h3>", unsafe_allow_html=True)
             st.write("Good news! The model predicts that there is no cancer detected. Stay healthy!")
     else:
-        st.error("Please train a model first!")
+        st.error("Please train a model first!")