Update app.py

app.py

@@ -46,10 +46,10 @@ MAX_SEED = np.iinfo(np.int32).max
 
 @spaces.GPU
 def run(prompt="a photo of an astronaut riding a horse on mars",
-        steps=10,
-        seed=20,
         negative_prompt="",
+        num_inference_steps=10,
         randomize_seed=False,
+        seed=20,
         progress=gr.Progress(track_tqdm=True)
     ):
     if randomize_seed:
@@ -60,6 +60,7 @@ def run(prompt="a photo of an astronaut riding a horse on mars",
     ays_images = pipe(
         prompt,
         negative_prompt=negative_prompt,
+        num_inference_steps=num_inference_steps,
         num_images_per_prompt=1,
         timesteps=sampling_schedule,
     ).images
@@ -129,7 +130,7 @@ with gr.Blocks(css=css) as demo:
     gr.on(
         [run_button.click, prompt.submit, negative_prompt.submit],
         fn = run,
-        inputs = [prompt, num_inference_steps, seed, negative_prompt, randomize_seed],
+        inputs = [prompt, negative_prompt, num_inference_steps, randomize_seed, seed],
         outputs = [result, seed]
     )