Update gradio_app.py

gradio_app.py

@@ -11,6 +11,13 @@ import shutil
 
 class DrugGENConfig:
     submodel='DrugGEN'
+    inference_model="experiments/models/DrugGEN/"
+    sample_num=1000
+    inf_dataset_file="chembl45_test.pt"
+    inf_raw_file='data/chembl_test.smi'
+    inf_batch_size=1
+    mol_data_dir='data'
+    features=False
     act='relu'
     max_atom=45
     dim=32
@@ -18,35 +25,15 @@ class DrugGENConfig:
     heads=8
     mlp_ratio=3
     dropout=0.
-    features=False
-    sample_num=1000
-    inf_batch_size=1
-    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="experiments/models/DrugGEN/"
-    sample_dir='experiments/samples'
-    result_dir="experiments/tboard_output"
-    dataset_file="chembl45_train.pt"
-    drug_dataset_file="akt_train.pt"
-    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='data/chembl_test.smi'
-    inf_drug_raw_file="data/akt_test.smi"
     log_sample_step=1000
-    set_seed=False
-    seed=1
+    set_seed=True
+    seed=10
 
 
 class NoTargetConfig(DrugGENConfig):
     submodel="NoTarget"
     dim=128
-    inference_model="experiments/models/NoTarget"
+    inference_model="experiments/models/NoTarget/"
 
 
 model_configs = {