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(
|