Update app.py

app.py

@@ -10,7 +10,6 @@ seed=42
 
 annotations = pd.read_csv("annotations_dataset.csv")
 annotations = annotations.set_index("Gene")
-annotations.rename_axis(index='Gene', inplace=True)
 
 training_data = pd.read_csv("./selected_features_training_data.csv", header=0)
 training_data.columns = [
@@ -68,6 +67,7 @@ def convert_df(df):
 
 if len(gene_list) > 1:
     df = df_total[df_total.index.isin(gene_list)]
+    df.rename_axis(index='Gene', inplace=True)
     st.dataframe(df)
     df['Gene'] = df.index
     output = df[['Gene', 'XGB_Score']]