From 9364ae07aa07628aadaa391aafb8039566789b4e Mon Sep 17 00:00:00 2001
From: Jesselyn Mu <90194068+jesselynmu@users.noreply.github.com>
Date: Thu, 30 Jan 2025 21:13:30 +0700
Subject: [PATCH] layout

---
 pages/prediction.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pages/prediction.py b/pages/prediction.py
index 02920e6..c553001 100644
--- a/pages/prediction.py
+++ b/pages/prediction.py
@@ -428,7 +428,7 @@ def show_prediction():
         plot_placeholder = st.empty()
 
         # Modify the generate_shap_plot function
-        def generate_shap_plot(X_test_class, explainer_class):
+        def generate_shap_plot(X_test_class, explainer_class, shap_dict, predicted_class):
             plt.close('all')
             
             try: