Update app.py

app.py

@@ -92,8 +92,8 @@ def run_inference(mode: str, model_name: str, num_molecules: int, seed_num: str,
     else:
         # Classical Generation mode
         config.sample_num = num_molecules
-        if config.sample_num > 200:
-            raise gr.Error("You have requested to generate more than the allowed limit of 200 molecules. Please reduce your request to 200 or fewer.")
+        if config.sample_num > 250:
+            raise gr.Error("You have requested to generate more than the allowed limit of 250 molecules. Please reduce your request to 250 or fewer.")
         if seed_num is None or seed_num.strip() == "":
             config.seed = random.randint(0, 10000)
         else:
@@ -276,11 +276,11 @@ For more details, see our [paper on arXiv](https://arxiv.org/abs/2302.07868).
 
                     num_molecules = gr.Slider(
                         minimum=10,
-                        maximum=200,
+                        maximum=250,
                         value=100,
                         step=10,
                         label="Number of Molecules to Generate",
-                        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."
+                        info="This space runs on a CPU, which may result in slower performance. Generating 200 molecules takes approximately 6 minutes. Therefore, we set a 250-molecule cap."
                     )
 
                     seed_num = gr.Textbox(
@@ -359,7 +359,7 @@ For more details, see our [paper on arXiv](https://arxiv.org/abs/2302.07868).
     # Set up the click actions for each tab.
     classical_submit.click(
         run_inference, 
-        inputs=[gr.State("Generate Molecules"), model_name, num_molecules, seed_num, gr.State("")],
+        inputs=[gr.Variable("Generate Molecules"), model_name, num_molecules, seed_num, gr.Textbox.update(value="")],
         outputs=[
             image_output, 
             file_download,
@@ -368,10 +368,10 @@ For more details, see our [paper on arXiv](https://arxiv.org/abs/2302.07868).
         ],
         api_name="inference_classical"
     )
-    
+
     custom_submit.click(
         run_inference, 
-        inputs=[gr.State("Custom Input SMILES"), model_name_custom, gr.State(0), gr.State(""), custom_smiles],
+        inputs=[gr.Variable("Custom Input SMILES"), model_name_custom, 0, gr.Textbox.update(value=""), custom_smiles],
         outputs=[
             image_output_custom, 
             file_download_custom,