Update app.py

app.py

@@ -26,12 +26,14 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
     image = pipe(
         prompt = prompt, 
         negative_prompt = negative_prompt,
+        width=width,
+        height=height,
         guidance_scale = guidance_scale, 
         num_inference_steps = num_inference_steps, 
         generator = generator
     ).images[0] 
     
-    return image
+    return image, seed
 
 examples = [
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
@@ -138,7 +140,7 @@ with gr.Blocks(css=css) as demo:
         triggers=[run_button.click, prompt.submit, negative_prompt.submit],
         fn = infer,
         inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
-        outputs = [result]
+        outputs = [result, seed]
     )
 
 demo.queue().launch()