Update app.py

app.py

@@ -14,8 +14,8 @@ explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
 
 # Create the main function for server
 def main_func(Target, Admission_grade, Curricular_units_2nd_sem_grade, Previous_qualifications, Curricular_units_1st_sem_grade, Curricular_units_2nd_sem_approved, Age_at_enrollment):
-    new_row = pd.DataFrame.from_dict({'Target':Target,'Curricular units 2nd sem (grade)':Curricular_units_2nd_sem_grade,
-              'Previous qualifications':Previous_qualifications,'Curricular units 1st sem (grade)':Curricular_units_1st_sem_grade,'Curricular units 2nd sem (approved)':Curricular_units_2nd_sem_approved,
+    new_row = pd.DataFrame.from_dict({'Target':Target, 'Admission grade':Admission_grade,'Curricular units 2nd sem (grade)':Curricular_units_2nd_sem_grade,
+              'Previous qualifications':Previous_qualifications,'Curricular units 1st sem (grade)':Curricular_units_1st_sem_grade, 'Course':Course,'Curricular units 2nd sem (approved)':Curricular_units_2nd_sem_approved,
               'Age at enrollment':Age_at_enrollmentenrollment}).transpose() 
     
     prob = loaded_model.predict_proba(new_row)
@@ -46,21 +46,13 @@ with gr.Blocks(title=title) as demo:
     gr.Markdown(description2)
     gr.Markdown("""---""")
     
-    age = gr.Number(label="age Score", value=40)
-    sex = gr.Slider(label="sex Score", minimum=0, maximum=1, value=1, step=1)
-    cp = gr.Slider(label="cp Score", minimum=1, maximum=5, value=4, step=1)
-    trtbps = gr.Slider(label="trtbps Score", minimum=1, maximum=5, value=4, step=1) 
-    chol = gr.Slider(label="chol Score", minimum=1, maximum=5, value=4, step=1)
-    fbs = gr.Slider(label="fbs Score", minimum=1, maximum=5, value=4, step=1)
-
-    restecg = gr.Slider(label="restecg Score", minimum=1, maximum=5, value=4, step=1)
-    thalachh = gr.Slider(label="thalachh Score", minimum=1, maximum=5, value=4, step=1)
-   
-    exng = gr.Slider(label="exng Score", minimum=1, maximum=5, value=4, step=1)
-    oldpeak = gr.Slider(label="oldpeak Score", minimum=1, maximum=5, value=4, step=1)
-    slp = gr.Slider(label="slp Score", minimum=1, maximum=5, value=4, step=1)
-    caa = gr.Slider(label="caa Score", minimum=1, maximum=5, value=4, step=1)
-    thall = gr.Slider(label="thall Score", minimum=1, maximum=5, value=4, step=1)
+    Curricular_units_2nd_sem_grade = gr.Number(label="Curricular units 2nd sem (grade)", value=40)
+    Admission_grade = gr.Slider(label="Admission grade", minimum=0, maximum=1, value=1, step=1)
+    Previous_qualifications = gr.Slider(label="Previous qualifications", minimum=1, maximum=5, value=4, step=1)
+    Curricular_units_1st_sem_grade = gr.Slider(label="Curricular units 1st sem (grade)", minimum=1, maximum=5, value=4, step=1) 
+    Course = gr.Slider(label="Course", minimum=1, maximum=5, value=4, step=1) 
+    Curricular_units_2nd_sem_approved = gr.Slider(label="Curricular units 2nd sem (approved)", minimum=1, maximum=5, value=4, step=1)
+    Age at enrollment = gr.Slider(label="Age_at_enrollmentenrollment", minimum=1, maximum=5, value=4, step=1)
     
     submit_btn = gr.Button("Analyze")
 
@@ -71,7 +63,7 @@ with gr.Blocks(title=title) as demo:
     submit_btn.click(
         main_func,
         [age, sex, cp, trtbps, chol, fbs, restecg, thalachh,exng,oldpeak,slp,caa,thall],
-        [label,local_plot], api_name="Heart_Predictor"
+        [label,local_plot], api_name="Dropout_Predictor"
     )
     
     gr.Markdown("### Click on any of the examples below to see how it works:")