Update app.py

app.py

@@ -9,25 +9,25 @@ with open('heart_disease_model.pkl', 'rb') as file:
 # Define the prediction function
 def predict_heart_disease(male, age, currentSmoker, cigsPerDay, BPMeds, prevalentStroke, prevalentHyp, diabetes, BMI):
     male = 1 if male == "Male" else 0
-    currentSmoker = 1 if currentSmoker == "Yes" else 0
-    BPMeds = 1 if BPMeds == "Yes" else 0
-    prevalentStroke = 1 if prevalentStroke == "Yes" else 0
-    prevalentHyp = 1 if prevalentHyp == "Yes" else 0
-    diabetes = 1 if diabetes == "Yes" else 0
+    currentSmoker = 1 if currentSmoker else 0
+    BPMeds = 1 if BPMeds else 0
+    prevalentStroke = 1 if prevalentStroke else 0
+    prevalentHyp = 1 if prevalentHyp else 0
+    diabetes = 1 if diabetes else 0
     input_data = np.array([[male, age, currentSmoker, cigsPerDay, BPMeds, prevalentStroke, prevalentHyp, diabetes, BMI]])
     prediction = model.predict(input_data)
     return 'Heart Disease' if prediction[0] == 1 else 'No Heart Disease'
 
 # Define the input components
 inputs = [
-    gr.Textbox(label="Sex (Male/Female)"),
+    gr.Radio(choices=["Male","Female"], label="Sex"),
     gr.Number(label="Age"),
-    gr.Text(label="Do you smoke? (Yes/No)"),
+    gr.Checkbox(label="Do you smoke?"),
     gr.Number(label="Number of ciggerates a day"),
-    gr.Textbox(label="Do you take BP medicines? (Yes/No)"),
-    gr.Textbox(label="Have you had strokes previously?(Yes/No)"),
-    gr.Textbox(label="Have you had High BP previously?(Yes/no)"),
-    gr.Textbox(label="Do you have Diabetes?(Yes/No)"),
+    gr.Checkbox(label="Do you take BP medicines?"),
+    gr.Checkbox(label="Have you had strokes previously?"),
+    gr.Checkbox(label="Have you had High BP previously?"),
+    gr.Checkbox(label="Do you have Diabetes?"),
     gr.Number(value=float, label="BMI")
 
 ]