Update app.py

app.py

@@ -24,6 +24,8 @@ pipe = FluxPipeline.from_pretrained(
 
 @spaces.GPU
 def infer(control_image, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
     pipe_prior_output = pipe_prior_redux(control_image)
     images = pipe(
         guidance_scale=guidance_scale,
@@ -31,11 +33,12 @@ def infer(control_image, seed=42, randomize_seed=False, width=1024, height=1024,
         generator=torch.Generator("cpu").manual_seed(seed),
         **pipe_prior_output,
     ).images[0]
-    return images
+    return images, seed
+
 css="""
 #col-container {
     margin: 0 auto;
-    max-width: 520px;
+    max-width: 600px;
 }
 """