varshitha22 commited on
Commit
671eb53
·
verified ·
1 Parent(s): 872d1e9

Update cancer.py

Browse files
Files changed (1) hide show
  1. cancer.py +6 -1
cancer.py CHANGED
@@ -101,7 +101,11 @@ if st.button("Predict Cancer Presence"):
101
  # Create DataFrame for input
102
  input_df = pd.DataFrame(input_data, columns=x_train.columns)
103
 
104
- # Apply preprocessing (fix error)
 
 
 
 
105
  input_transformed = model.named_steps['preprocessor'].transform(input_df)
106
 
107
  # Make prediction
@@ -115,3 +119,4 @@ if st.button("Predict Cancer Presence"):
115
  st.write("Good news! The model predicts that there is no cancer detected. Stay healthy!")
116
  else:
117
  st.error("Please train a model first!")
 
 
101
  # Create DataFrame for input
102
  input_df = pd.DataFrame(input_data, columns=x_train.columns)
103
 
104
+ # Convert numeric inputs explicitly to float
105
+ for col in ['Age', 'Tumor_Size']:
106
+ input_df[col] = pd.to_numeric(input_df[col], errors='coerce')
107
+
108
+ # Apply preprocessing
109
  input_transformed = model.named_steps['preprocessor'].transform(input_df)
110
 
111
  # Make prediction
 
119
  st.write("Good news! The model predicts that there is no cancer detected. Stay healthy!")
120
  else:
121
  st.error("Please train a model first!")
122
+