Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -74,6 +74,8 @@ def function(model_name: str, num_molecules: int, seed_num: int):
|
|
| 74 |
'''
|
| 75 |
if model_name == "DrugGEN-NoTarget":
|
| 76 |
model_name = "NoTarget"
|
|
|
|
|
|
|
| 77 |
|
| 78 |
config = model_configs[model_name]
|
| 79 |
config.sample_num = num_molecules
|
|
|
|
| 74 |
'''
|
| 75 |
if model_name == "DrugGEN-NoTarget":
|
| 76 |
model_name = "NoTarget"
|
| 77 |
+
else:
|
| 78 |
+
model_name = "DrugGEN"
|
| 79 |
|
| 80 |
config = model_configs[model_name]
|
| 81 |
config.sample_num = num_molecules
|