Update app.py

app.py

@@ -125,6 +125,10 @@ def explainability(_):
         else:
             sv = shap_values
 
+        # Ensure 2D input shape for SHAP plot
+        if len(sv.shape) > 2:
+            sv = sv.reshape(sv.shape[0], -1)  # Flatten any extra dimensions
+
         # Use safe feature names if mismatch, fallback to dummy
         num_features = sv.shape[1]
         if num_features <= X_test.shape[1]:
@@ -173,6 +177,7 @@ def explainability(_):
 
 
 
+
 with gr.Blocks() as demo:
     gr.Markdown("## 📊 AI-Powered Data Analysis with Hyperparameter Optimization")