Update

app.py

@@ -234,9 +234,10 @@ def create_demo(model_name: str, device: str = "cuda", offload: bool = False):
                     with gr.Row():
                         seed = gr.Textbox(-1, label="Seed (-1 for random)")
                         img_size = gr.Radio(["1,024x1,024", "512x512"], label="Image Resolution", value="1,024x1,024")
-                    num_steps = gr.Slider(1, 4, 1, step=1, label="Number of steps")
+                    num_steps = gr.Slider(1, 50, 4 if is_schnell else 50, step=1, label="Number of steps")
+                    guidance = gr.Slider(1.0, 10.0, 3.5, step=0.1, label="Guidance", interactive=not is_schnell)
                     add_sampling_metadata = gr.Checkbox(label="Add sampling parameters to metadata?", value=True)
-                    guidance = gr.Slider(1.0, 10.0, 3.5, step=0.1, label="Guidance", interactive=not is_schnell, visible=False)
+                    # guidance = gr.Slider(1.0, 10.0, 3.5, step=0.1, label="Guidance", interactive=not is_schnell, visible=False)
                 
             
 
@@ -265,5 +266,5 @@ def create_demo(model_name: str, device: str = "cuda", offload: bool = False):
 #     parser.add_argument("--share", action="store_true", help="Create a public link to your demo")
 #     args = parser.parse_args()
 
-demo = create_demo("flux-schnell", None, False)
+demo = create_demo("flux-dev", None, False)
 demo.launch()