Update app.py

app.py

@@ -40,23 +40,23 @@ class DrugGENConfig:
     mode="inference"
     inference_iterations=100
     inf_batch_size=1
-    protein_data_dir='DrugGEN/data/akt'
-    drug_index='DrugGEN/data/drug_smiles.index'
-    drug_data_dir='DrugGEN/data/akt'
-    mol_data_dir='DrugGEN/data'
-    log_dir='DrugGEN/experiments/logs'
-    model_save_dir='DrugGEN/experiments/models'
+    protein_data_dir='data/akt'
+    drug_index='data/drug_smiles.index'
+    drug_data_dir='data/akt'
+    mol_data_dir='data'
+    log_dir='experiments/logs'
+    model_save_dir='experiments/models'
     inference_model=""
-    sample_dir='DrugGEN/experiments/samples'
-    result_dir="DrugGEN/experiments/tboard_output"
+    sample_dir='experiments/samples'
+    result_dir="experiments/tboard_output"
     dataset_file="chembl45_train.pt"
     drug_dataset_file="akt_train.pt"
-    raw_file='DrugGEN/data/chembl_train.smi'
-    drug_raw_file="DrugGEN/data/akt_train.smi"
+    raw_file='data/chembl_train.smi'
+    drug_raw_file="data/akt_train.smi"
     inf_dataset_file="chembl45_test.pt"
     inf_drug_dataset_file='akt_test.pt'
-    inf_raw_file='DrugGEN/data/chembl_test.smi'
-    inf_drug_raw_file="DrugGEN/data/akt_test.smi"
+    inf_raw_file='data/chembl_test.smi'
+    inf_drug_raw_file="data/akt_test.smi"
 
     
 with st.spinner('Setting up the trainer class...'):