fix layout size

app.py

@@ -146,10 +146,11 @@ with gr.Blocks() as demo:
         )
 
         fig2 = plt.figure(figsize=(12, 6))  # Create a new figure for SHAP plot
+        plt.title("SHAP Waterfall Plot")  # Optionally set a title for the SHAP plot
+        plt.tight_layout()
         shap.waterfall_plot(
             shap_explanation
         )  # Set show=False to prevent immediate display
-        plt.title("SHAP Waterfall Plot")  # Optionally set a title for the SHAP plot
 
         ## save user's data in hopsworks
         if consent_input == True: