Update app.py

app.py

@@ -67,9 +67,9 @@ if len(gene_list) > 1:
     st.dataframe(df)
     df.drop(columns='XGB_Score', inplace=True)
     shap_values = explainer.shap_values(df)
-    summary_plot = shap.summary_plot(shap_values, df)
+    summary_plot = shap.summary_plot(shap_values, df, display = False)
     st.caption("SHAP Summary Plot of All Input Genes")
-    components.html(summary_plot, height = output_height, width = output_width, scrolling = True)
+    components.html(summary_plot, scrolling = True)
 
 else:
     pass
@@ -85,8 +85,9 @@ if len(input_gene) == 1:
     force_plot = shap.force_plot(
     explainer.expected_value,
     shap_values.values,
-    df2)
-    components.html(force_plot, height = output_height, width = output_width, scrolling = True)
+    df2,
+    display = False)
+    components.html(force_plot,  scrolling = True)
 else:
     pass