Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -92,8 +92,8 @@ def run_inference(mode: str, model_name: str, num_molecules: int, seed_num: str,
|
|
| 92 |
else:
|
| 93 |
# Classical Generation mode
|
| 94 |
config.sample_num = num_molecules
|
| 95 |
-
if config.sample_num >
|
| 96 |
-
raise gr.Error("You have requested to generate more than the allowed limit of
|
| 97 |
if seed_num is None or seed_num.strip() == "":
|
| 98 |
config.seed = random.randint(0, 10000)
|
| 99 |
else:
|
|
@@ -276,11 +276,11 @@ For more details, see our [paper on arXiv](https://arxiv.org/abs/2302.07868).
|
|
| 276 |
|
| 277 |
num_molecules = gr.Slider(
|
| 278 |
minimum=10,
|
| 279 |
-
maximum=
|
| 280 |
value=100,
|
| 281 |
step=10,
|
| 282 |
label="Number of Molecules to Generate",
|
| 283 |
-
info="This space runs on a CPU, which may result in slower performance. Generating
|
| 284 |
)
|
| 285 |
|
| 286 |
seed_num = gr.Textbox(
|
|
|
|
| 92 |
else:
|
| 93 |
# Classical Generation mode
|
| 94 |
config.sample_num = num_molecules
|
| 95 |
+
if config.sample_num > 200:
|
| 96 |
+
raise gr.Error("You have requested to generate more than the allowed limit of 200 molecules. Please reduce your request to 200 or fewer.")
|
| 97 |
if seed_num is None or seed_num.strip() == "":
|
| 98 |
config.seed = random.randint(0, 10000)
|
| 99 |
else:
|
|
|
|
| 276 |
|
| 277 |
num_molecules = gr.Slider(
|
| 278 |
minimum=10,
|
| 279 |
+
maximum=200,
|
| 280 |
value=100,
|
| 281 |
step=10,
|
| 282 |
label="Number of Molecules to Generate",
|
| 283 |
+
info="This space runs on a CPU, which may result in slower performance. Generating 100 molecules takes approximately 6 minutes. Therefore, we set a 200-molecule cap."
|
| 284 |
)
|
| 285 |
|
| 286 |
seed_num = gr.Textbox(
|