Spaces:
Sleeping
Sleeping
criticalDanger
commited on
Commit
•
7b5b571
1
Parent(s):
b0ac891
Update app.py
Browse files
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 = [
|