Minor fix

app.py

@@ -4,8 +4,8 @@ from os import listdir
 import plotly.graph_objects as go
 
 gene_names_eve = [file.split('_')[0] for file in listdir('./CPT1_score_EVE_set/')]
-gene_names_no_eve1 = [file.split('_')[0] for file in listdir('./CPT1_score_no_EVE_set1/')]
-gene_names_no_eve2 = [file.split('_')[0] for file in listdir('./CPT1_score_no_EVE_set2/')]
+gene_names_no_eve1 = [file.split('_')[0] for file in listdir('./CPT1_score_no_EVE_set_1/')]
+gene_names_no_eve2 = [file.split('_')[0] for file in listdir('./CPT1_score_no_EVE_set_2/')]
 
 st.subheader('CPT-1')
 st.markdown('Cross-protein transfer learning for variant effect prediction')
@@ -21,9 +21,9 @@ if st.button('Show results'):
     if gene in gene_names_eve:
         pred = pd.read_csv('./CPT1_score_EVE_set/' + gene + '_HUMAN.csv.gz', compression = 'gzip')
     elif gene in gene_names_no_eve1:
-        pred = pd.read_csv('./CPT1_score_no_EVE_set1/' + gene + '_HUMAN.csv.gz', compression = 'gzip')
+        pred = pd.read_csv('./CPT1_score_no_EVE_set_1/' + gene + '_HUMAN.csv.gz', compression = 'gzip')
     else:
-        pred = pd.read_csv('./CPT1_score_no_EVE_set2/' + gene + '_HUMAN.csv.gz', compression = 'gzip')
+        pred = pd.read_csv('./CPT1_score_no_EVE_set_2/' + gene + '_HUMAN.csv.gz', compression = 'gzip')
     
     # Plot heatmap
     mat = pred.copy()