Update app.py

app.py

@@ -134,10 +134,11 @@ if len(gene_list) > 1:
     st.caption("SHAP Summary Plot of All Input Genes")
     st.pyplot(fig=summary_plot)
     st.caption("Interactive SHAP Summary Plot of All Input Genes")
-    mpl_fig = shap_summary_plot(shap_values, df_shap, max_display=8, show=False)
+    mpl_fig = shap_summary_plot(shap_values, df_shap, max_display=8, show=False, feature_names=df_shap.columns)
     plotly_fig = tls.mpl_to_plotly(mpl_fig)
     plotly_fig['layout'] = {'xaxis': {'title': 'SHAP value (impact on model output)'}}
     max_display=8
+    feature_names=df_shap.columns
     feature_order = np.argsort(np.sum(np.abs(shap_values), axis=0)[:-1])
     feature_order = feature_order[-min(max_display, len(feature_order)):]
     text = [feature_names[i] for i in feature_order]