Back to default

app.py

@@ -23,7 +23,7 @@ torch.cuda.empty_cache()
 
 # Inference function
 @spaces.GPU(duration=25)
-def generate_image(prompt, seed=42, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT, num_inference_steps=DEFAULT_INFERENCE_STEPS, randomize_seed=False):
+def generate_image(prompt, seed=42, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT, randomize_seed=False, num_inference_steps=DEFAULT_INFERENCE_STEPS):
     
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -108,7 +108,7 @@ with gr.Blocks() as demo:
     gr.on(
         triggers=[prompt.input, width.input, height.input, num_inference_steps.input],
         fn=generate_image,
-        inputs=[prompt, seed, width, height, num_inference_steps, randomize_seed],
+        inputs=[prompt, seed, width, height, randomize_seed, num_inference_steps],
         outputs=[result, seed, latency],
         show_progress="hidden",
         show_api=False,
@@ -117,4 +117,4 @@ with gr.Blocks() as demo:
     )
 
 # Launch the app
-demo.launch() 
+demo.launch()