shap

app.py

@@ -115,7 +115,13 @@ with gr.Blocks() as demo:
         explainer = shap.Explainer(rf_model)
         shap_values = explainer.shap_values(df)[1]  # Select SHAP values for class 1
         print(shap_values.shape)  # should show (1, 4)
-        shap_waterfall_plot = shap.plots.waterfall(shap_values[0])
+        # Convert shap_values to Explanation object
+        shap_values_exp = shap.Explanation(
+            values=shap_values[0], base_values=explainer.expected_value[1]
+        )
+
+        # Plot the waterfall plot
+        shap_waterfall_plot = shap.plots.waterfall(shap_values_exp)
 
         ## save user's data in hopsworks
         if consent_input == True: