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.Variable("Generate Molecules"), model_name, num_molecules, seed_num, gr.Textbox.update(value="")],
+        inputs=[gr.State("Generate Molecules"), model_name, num_molecules, seed_num, gr.Textbox.update(value="")],
         outputs=[
             image_output, 
             file_download,
@@ -371,7 +371,7 @@ For more details, see our [paper on arXiv](https://arxiv.org/abs/2302.07868).
 
     custom_submit.click(
         run_inference, 
-        inputs=[gr.Variable("Custom Input SMILES"), model_name_custom, 0, gr.Textbox.update(value=""), custom_smiles],
+        inputs=[gr.State("Custom Input SMILES"), model_name_custom, 0, gr.Textbox.update(value=""), custom_smiles],
         outputs=[
             image_output_custom, 
             file_download_custom,