Update app.py

app.py

@@ -770,7 +770,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
 
 # Reduce inference steps to save memory
-def generate_image_flux(prompt, seed=42, width=1024, height=1024, num_inference_steps=2):
+def generate_image_flux(prompt, seed=42, width=width, height=height, num_inference_steps=2):
     generator = torch.Generator(device).manual_seed(seed)
     image = pipe(
         prompt=prompt,