Update app.py

app.py

@@ -2,16 +2,22 @@ import gradio as gr
 from gradio_molecule3d import Molecule3D
 from gradio_cofoldinginput import CofoldingInput
 
-def predict(inputs):
+def predict(jobname, inputs, recycling_steps, sampling_steps, diffusion_samples):
     return ""
 
 with gr.Blocks() as blocks:
     gr.Markdown("# Boltz-1")
-    jobname = gr.Textbox(label="Textbox")
-    inp = CofoldingInput(label="Input")
-    out = Molecule3D(label="Output")
+    with gr.Tab("Main"):
+        jobname = gr.Textbox(label="Jobname")
+        inp = CofoldingInput(label="Input")
+        out = Molecule3D(label="Output")
+    with gr.Tab("Settings"):
+        recycling_steps =gr.Slider(value=3, minimum=0)
+        sampling_steps = gr.Slider(value=200, minimum=0)
+        diffusion_samples = gr.Slider(value=1)
+
     btn = gr.Button("predict")
 
-    btn.click(fn=predict, inputs=[jobname,inp], outputs=[out],  api_name="predict")
+    btn.click(fn=predict, inputs=[jobname,inp, recycling_steps, sampling_steps, diffusion_samples], outputs=[out],  api_name="predict")
 
 blocks.launch(ssr_mode=False)