1-2 step inference

app.py

@@ -97,9 +97,11 @@ with gr.Blocks() as demo:
                 width = gr.Slider(
                     label="Image Width", minimum=768, maximum=1024, step=8, value=1024, interactive=True
                 )
+                inference_steps = gr.Slider(
+                    label="Inference Steps", minimum=1, maximum=2, step=1, value=1, interactive=True,
+                )
                 prompt = gr.Text(label="Prompt", value="a photo of a cat", interactive=True)
                 seed = gr.Number(label="Seed", value=2024, interactive=True)
-                inference_steps = gr.Number(label="Inference Steps", value=1, interactive=False)
                 btn = gr.Button(value="Generate Image")
             with gr.Column():
                 output = gr.Gallery(height=1024)