Update app.py

app.py

@@ -29,8 +29,7 @@ xgb = xgboost.XGBRegressor(
     reg_alpha=1,
     reg_lambda=1,
     random_state=seed,
-    objective="reg:squarederror",
-)
+    objective="reg:squarederror")
 
 
 xgb.fit(X, Y)
@@ -69,7 +68,6 @@ def convert_df(df):
 if len(gene_list) > 1:
     df = df_total[df_total.index.isin(gene_list)]
     st.dataframe(df)
-    df_shap = df
     df['Gene'] = df.index
     output = df[['Gene', 'XGB_Score']]
     csv = convert_df(output)
@@ -80,6 +78,7 @@ if len(gene_list) > 1:
        "text/csv",
        key='download-csv'
     )
+    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)