fix plots

app.py

@@ -156,12 +156,8 @@ with gr.Blocks() as demo:
         plt.gca().set_position((0, 0, 1, 1))
         plt.title("SHAP Waterfall Plot")  # Optionally set a title for the SHAP plot
         plt.tight_layout()
-        # plt.xticks(rotation=90)
-        # plt.yticks(rotation=45)
         plt.tick_params(axis="y", labelsize=3)
-        shap.waterfall_plot(
-            shap_explanation
-        )  # Set show=False to prevent immediate display
+        shap.waterfall_plot(shap_explanation)
 
         fig3 = plt.figure(figsize=(3, 3))
         plt.title("SHAP Summary Plot")
@@ -171,7 +167,7 @@ with gr.Blocks() as demo:
             feature_names=["age", "bmi", "hba1c", "glucose"],
         )
 
-        fig4 = plt.figure(figsize=(3, 3))
+        fig4 = plt.figure(figsize=(4, 3))
         feature_importances = rf_classifier.feature_importances_
         plt.title("Feature Importances")
         sns.barplot(x=feature_importances, y=["age", "bmi", "hba1c", "glucose"])
@@ -208,7 +204,14 @@ with gr.Blocks() as demo:
             existent_info_input,
             consent_input,
         ],
-        outputs=[output, mean_plot, waterfall_plot, importance_plot, decision_plot],
+        outputs=[
+            output,
+            mean_plot,
+            waterfall_plot,
+            summary_plot,
+            importance_plot,
+            decision_plot,
+        ],
     )
 
 demo.launch()