criticalDanger commited on
Commit
7b5b571
1 Parent(s): b0ac891

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def predict_heart_disease(male, age, currentSmoker, cigsPerDay, BPMeds, prevalen
16
  diabetes = 1 if diabetes else 0
17
  input_data = np.array([[male, age, currentSmoker, cigsPerDay, BPMeds, prevalentStroke, prevalentHyp, diabetes, BMI]])
18
  prediction = model.predict(input_data)
19
- return 'Heart Disease' if prediction[0] == 1 else 'No Heart Disease'
20
 
21
  # Define the input components
22
  inputs = [
 
16
  diabetes = 1 if diabetes else 0
17
  input_data = np.array([[male, age, currentSmoker, cigsPerDay, BPMeds, prevalentStroke, prevalentHyp, diabetes, BMI]])
18
  prediction = model.predict(input_data)
19
+ return 'Possible risk of Heart Disease' if prediction[0] == 1 else 'No possible risk of Heart Disease'
20
 
21
  # Define the input components
22
  inputs = [