DasariHarshitha commited on
Commit
e882ed4
·
verified ·
1 Parent(s): b990dea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -64,6 +64,7 @@ def train_model(X_train, y_train, preprocess, model_name):
64
  'Logistic Regression': LogisticRegression(),
65
  'KNN': KNeighborsClassifier()
66
  }
 
67
  model = models[model_name]
68
  pipeline = Pipeline([
69
  ('preprocessor', preprocess),
@@ -113,4 +114,4 @@ if st.button("🔮 Predict Cancer Presence"):
113
 
114
  st.write("Cancer Prediction:", "✅ Positive" if prediction[0] == 1 else "❌ Negative")
115
  else:
116
- st.error("Please train a model first!")
 
64
  'Logistic Regression': LogisticRegression(),
65
  'KNN': KNeighborsClassifier()
66
  }
67
+
68
  model = models[model_name]
69
  pipeline = Pipeline([
70
  ('preprocessor', preprocess),
 
114
 
115
  st.write("Cancer Prediction:", "✅ Positive" if prediction[0] == 1 else "❌ Negative")
116
  else:
117
+ st.error("Please train a model first!")