Update app.py

app.py

@@ -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, ""],
+        inputs=[gr.State("Generate Molecules"), model_name, num_molecules, seed_num, gr.State("")],
         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, 0, "", custom_smiles],
+        inputs=[gr.State("Custom Input SMILES"), model_name_custom, gr.State(0), gr.State(""), custom_smiles],
         outputs=[
             image_output_custom, 
             file_download_custom,