Update app.py

app.py

@@ -35,6 +35,8 @@ MAX_IMAGE_SIZE = 2048
 
 @spaces.GPU()
 def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
     image = pipe(
             prompt = prompt, 
@@ -55,7 +57,7 @@ examples = [
 css="""
 #col-container {
     margin: 0 auto;
-    max-width: 520px;
+    max-width: 550px;
 }
 """
 
@@ -64,9 +66,8 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
         # FLUX.1 Schnell
-        12B parameters rectified flow transformer distilled from [FLUX.1 Pro](https://blackforestlabs.ai/) for fast 4 step image generation
-        
-        [[blog](https://blackforestlabs.ai/2024/07/31/announcing-black-forest-labs/)] [[model](https://black-forest-labs/FLUX.1-schnell)]]
+        12B param rectified flow transformer distilled from [FLUX.1 Pro](https://blackforestlabs.ai/) for 4 step generation
+        [[blog](https://blackforestlabs.ai/2024/07/31/announcing-black-forest-labs/) [model](https://black-forest-labs/FLUX.1-schnell)]
         """)
         
         with gr.Row():