Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -42,14 +42,14 @@ class DrugGENConfig:
|
|
42 |
|
43 |
class DrugGENAKT1Config(DrugGENConfig):
|
44 |
submodel='DrugGEN'
|
45 |
-
inference_model="/home/user/app/experiments/models/DrugGEN-
|
46 |
train_drug_smiles='/home/user/app/data/akt_train.smi'
|
47 |
max_atom=45
|
48 |
|
49 |
|
50 |
class DrugGENCDK2Config(DrugGENConfig):
|
51 |
submodel='DrugGEN'
|
52 |
-
inference_model="/home/user/app/experiments/models/DrugGEN-
|
53 |
train_drug_smiles='/home/user/app//data/cdk2_train.smi'
|
54 |
max_atom=38
|
55 |
|
|
|
42 |
|
43 |
class DrugGENAKT1Config(DrugGENConfig):
|
44 |
submodel='DrugGEN'
|
45 |
+
inference_model="/home/user/app/experiments/models/DrugGEN-akt1/"
|
46 |
train_drug_smiles='/home/user/app/data/akt_train.smi'
|
47 |
max_atom=45
|
48 |
|
49 |
|
50 |
class DrugGENCDK2Config(DrugGENConfig):
|
51 |
submodel='DrugGEN'
|
52 |
+
inference_model="/home/user/app/experiments/models/DrugGEN-cdk2/"
|
53 |
train_drug_smiles='/home/user/app//data/cdk2_train.smi'
|
54 |
max_atom=38
|
55 |
|