Spaces:
Sleeping
Sleeping
Update cancer.py
Browse files
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 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
|