Update gradio_app.py

gradio_app.py

@@ -57,7 +57,8 @@ def function(model_name: str, num_molecules: int, seed_num: int) -> tuple[PIL.Im
     config.sample_num = num_molecules
 
     if config.sample_num > 250:
-        raise gr.Error("LIMIT ERROR!")
+        raise gr.Error("You have requested to generate more than the allowed limit of 250 molecules. Please reduce your request to 250 or fewer molecules. \
+                            If you need to generate more molecules, please visit our GitHub repository for instructions on running the model on your own setup with no limit.")
 
     if seed_num is None or seed_num.strip() == "":
         config.seed = random.randint(0, 10000)
@@ -139,7 +140,7 @@ with gr.Blocks() as demo:
                 minimum=1,
                 value=1000,
                 maximum=10_000,
-                info="It will take ~20-60 seconds to generate 1000 mols."
+                info="This model currently runs on a CPU, which may result in slower performance. Generating 200 molecules takes approximately 6 minutes on this space. The model can generate 10,000 molecules on a GPU in the same amount of time. For faster results, consider running the model on a GPU."
             )
 
             seed_num = gr.Textbox(