Update app.py

app.py

@@ -134,7 +134,7 @@ if len(gene_list) > 1:
     st.caption("SHAP Summary Plot of All Input Genes")
     feature_order = np.argsort(np.sum(np.abs(shap_values), axis=0)[:-1])
     feature_order = feature_order[-min(8, len(feature_order)):]
-    col_order = [df.columns[i] for i in feature_order]
+    col_order = [df_shap.columns[i] for i in feature_order]
     st.caption("Interactive SHAP Summary Plot of All Input Genes")
     mpl_fig = shap_summary_plot(shap_values, df_shap, max_display=8, show=False, feature_names=col_order)
     plotly_fig = tls.mpl_to_plotly(mpl_fig)