adjust graph, change ui model output

app.py

@@ -49,9 +49,9 @@ with gr.Blocks() as demo:
                 ["yes", "no", "Don't know"],
                 label="Do you already know if you have diabetes? (This will not be used for the prediction)",
             )
-            consent_input = gr.Checkbox(
-                info="I consent that my personal data will be saved and potentially be used for the model training",
-                label="accept",
+            consent_input = gr.Radio(
+                ["accept", "decline"],
+                label="I consent that my personal data will be saved and potentially be used for the model training",
             )
             btn = gr.Button("Submit")
         with gr.Column():
@@ -85,7 +85,10 @@ with gr.Blocks() as demo:
             columns=["age", "bmi", "hba1c_level", "blood_glucose_level"],
         )
         res = model.predict(df)
-
+        if res == [0]:
+            res = "the model prediction is: You don't have diabetes"
+        elif res == [1]:
+            res = "the model prediction is: You have diabetes"
         mean_for_age = diabetes_df[
             (diabetes_df["diabetes"] == 0) & (diabetes_df["age"] == age_input)
         ].mean()
@@ -173,7 +176,10 @@ with gr.Blocks() as demo:
         sns.barplot(x=feature_importances, y=["age", "bmi", "hba1c", "glucose"])
 
         fig5 = plt.figure(figsize=(3, 3))
+        fig5.tight_layout()
+        plt.gca().set_position((0, 0, 1, 1))
         plt.title("SHAP Interaction Plot")
+        plt.tight_layout()
         shap.decision_plot(
             explainer.expected_value[predicted_class],
             shap_values_for_predicted_class,
@@ -181,7 +187,7 @@ with gr.Blocks() as demo:
         )
 
         ## save user's data in hopsworks
-        if consent_input == True:
+        if consent_input == "accept":
             user_data_fg = fs.get_or_create_feature_group(
                 name="user_diabetes_data",
                 version=1,