Update app.py

app.py

@@ -639,7 +639,7 @@ with EmptyInitWrapper():
 @spaces.GPU(duration=70)
 @torch.no_grad()
 def generate_image(
-    prompt, neg_prompt, width, height, guidance, seed,
+    prompt, neg_prompt,num_steps ,width, height, guidance, seed,
     do_img2img, init_image, image2image_strength, resize_img,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -664,7 +664,7 @@ def generate_image(
     generator = torch.Generator(device=device).manual_seed(seed)
     x = torch.randn(1, 16, 2 * math.ceil(height / 16), 2 * math.ceil(width / 16), device=device, dtype=torch.bfloat16, generator=generator) 
     
-    num_steps = 28
+    # num_steps = 28
     timesteps = get_schedule(num_steps, (x.shape[-1] * x.shape[-2]) // 4, shift=True)
 
     if do_img2img and init_image is not None:
@@ -696,6 +696,7 @@ def create_demo():
             with gr.Column():
                 prompt = gr.Textbox(label="Prompt", value="a photo of a forest with mist swirling around the tree trunks. The word 'FLUX' is painted over it in big, red brush strokes with visible texture")
                 neg_prompt = gr.Textbox(label="Negative Prompt", value="bad photo")
+                num_steps = gr.Slider(minimum=1, maximum=30, step=10, label="num_steps", value=10)
                 width = gr.Slider(minimum=128, maximum=2048, step=64, label="Width", value=1360)
                 height = gr.Slider(minimum=128, maximum=2048, step=64, label="Height", value=768)
                 guidance = gr.Slider(minimum=1.0, maximum=5.0, step=0.1, label="Guidance", value=3.5)
@@ -718,7 +719,7 @@ def create_demo():
 
         generate_button.click(
             fn=generate_image,
-            inputs=[prompt, neg_prompt, width, height, guidance, seed, do_img2img, init_image, image2image_strength, resize_img],
+            inputs=[prompt, neg_prompt, num_steps,width, height, guidance, seed, do_img2img, init_image, image2image_strength, resize_img],
             outputs=[output_image, output_seed]
         )