Update app.py

app.py

@@ -130,11 +130,11 @@ if len(gene_list) > 1:
     df_shap = df_total[df_total.index.isin(gene_list)]
     df_shap.drop(columns='XGB_Score', inplace=True)
     shap_values = explainer.shap_values(df_shap)
-    summary_plot = shap.summary_plot(shap_values, df_shap, show=False)
+    summary_plot = shap.summary_plot(shap_values, df_shap)
     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, show=False)
+    mpl_fig = shap_summary_plot(shap_values, df_shap, max_display=20, show=False)
     plotly_fig = tls.mpl_to_plotly(mpl_fig)
     plotly_fig['layout'] = {'xaxis': {'title': 'SHAP value (impact on model output)'}}