width n height

app.py

@@ -2,6 +2,7 @@ import random
 import os
 
 import spaces
+import numpy as np
 import torch
 from PIL import Image
 import huggingface_hub
@@ -11,6 +12,10 @@ from src.pipeline_flux_nag import NAGFluxPipeline
 from src.transformer_flux import NAGFluxTransformer2DModel
 
 
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 2048
+
+
 theme = gr.themes.Base(
     font=[gr.themes.GoogleFont('Libre Franklin'), gr.themes.GoogleFont('Public Sans'), 'system-ui', 'sans-serif'],
 )
@@ -31,8 +36,7 @@ pipe = pipe.to(device)
 
 examples = [
     ["Portrait of AI researcher.", "Glasses.", 5],
-    ["A beautiful cyborg.", "Robot.", 5],
-    ["Minimalist abstract line drawing: face portrait of a girl with long hair.", "Complex, detail.", 7],
+    ["Portrait of AI researcher.", "Male.", 5],
     ["A baby phoenix made of fire and flames is born from the smoking ashes.", "Low resolution, blurry, lack of details, illustration, cartoon, painting.", 5],
     ["A tiny astronaut hatching from an egg on the moon.", "Low resolution, blurry, lack of details, illustration, cartoon, painting.", 9]
 ]
@@ -42,6 +46,7 @@ def sample(
         prompt,
         negative_prompt=None, guidance_scale=3.5,
         nag_negative_prompt=None, nag_scale=5.0,
+        width=1024, height=1024,
         num_inference_steps=25,
         seed=2025, randomize_seed=False,
         compare=True,
@@ -49,10 +54,11 @@ def sample(
     prompt = prompt.strip()
     negative_prompt = negative_prompt.strip() if negative_prompt and negative_prompt.strip() else None
     guidance_scale = float(guidance_scale)
+    width, height = int(width), int(height)
     num_inference_steps = int(num_inference_steps)
     
     if (randomize_seed):
-        seed = random.randint(0, 9007199254740991)
+        seed = random.randint(0, MAX_SEED)
     else:
         seed = int(seed)
 
@@ -64,6 +70,8 @@ def sample(
         nag_negative_prompt=nag_negative_prompt,
         nag_scale=nag_scale,
         generator=generator,
+        width=width,
+        height=height,
         num_inference_steps=num_inference_steps,
     ).images[0]
 
@@ -74,6 +82,8 @@ def sample(
             negative_prompt=negative_prompt,
             guidance_scale=guidance_scale,
             generator=generator,
+            width=width,
+            height=height,
             num_inference_steps=num_inference_steps,
         ).images[0]
     else:
@@ -92,7 +102,7 @@ def sample_example(
         nag_negative_prompt=nag_negative_prompt,
         nag_scale=nag_scale,
     )
-    return outputs, 3.5, 25, seed, True
+    return outputs, 3.5, 1024, 1024, 25, seed, True
 
 
 css = '''
@@ -124,8 +134,23 @@ with gr.Blocks(css=css, theme=theme) as demo:
         with gr.Accordion("Advanced Settings", open=False):
             negative_prompt = gr.Textbox(label="Negative Prompt", value=None, visible=False)
             guidance_scale = gr.Slider(label="Guidance Scale", minimum=1., maximum=15., step=0.1, value=3.5)
+            with gr.Row():
+                width = gr.Slider(
+                    label="Width",
+                    minimum=256,
+                    maximum=MAX_IMAGE_SIZE,
+                    step=32,
+                    value=1024,
+                )
+                height = gr.Slider(
+                    label="Height",
+                    minimum=256,
+                    maximum=MAX_IMAGE_SIZE,
+                    step=32,
+                    value=1024,
+                )
             num_inference_steps = gr.Slider(label="Inference Steps", minimum=1, maximum=50, step=1, value=25)
-            seed = gr.Slider(label="Seed", minimum=1, maximum=9007199254740991, step=1, randomize=True)
+            seed = gr.Slider(label="Seed", minimum=1, maximum=MAX_SEED, step=1, randomize=True)
             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
         gr.Examples(
@@ -136,7 +161,7 @@ with gr.Blocks(css=css, theme=theme) as demo:
                 nag_negative_prompt,
                 nag_scale,
             ],
-            outputs=[output, guidance_scale, num_inference_steps, seed, compare],
+            outputs=[output, guidance_scale, width, height, num_inference_steps, seed, compare],
             cache_examples="lazy",
         )
 
@@ -150,6 +175,7 @@ with gr.Blocks(css=css, theme=theme) as demo:
             prompt,
             negative_prompt, guidance_scale,
             nag_negative_prompt, nag_scale,
+            width, height,
             num_inference_steps,
             seed, randomize_seed,
             compare,