Update app.py

app.py

@@ -28,19 +28,19 @@ examples = json.loads(open("examples.json").read())
 # (1664, 928), (1472, 1140), (1328, 1328)
 def get_image_size(aspect_ratio):
     if aspect_ratio == "1:1":
-        return 1920, 1920
+        return 1328, 1328
     elif aspect_ratio == "16:9":
-        return 1920, 1080
+        return 1664, 928
     elif aspect_ratio == "9:16":
-        return 1080, 1920
+        return 928, 1664
     elif aspect_ratio == "4:3":
-        return 1920, 1440
+        return 1472, 1140
     elif aspect_ratio == "3:4":
-        return 1440, 1920
+        return 1140, 1472
     else:
-        return 640, 640
+        return 1328, 1328
 
-@spaces.GPU(duration=45)
+@spaces.GPU(duration=60)
 
 def infer(
     prompt,
@@ -52,21 +52,22 @@ def infer(
     num_inference_steps=50,
     progress=gr.Progress(track_tqdm=True),
 ):
-  
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
+        
     width, height = get_image_size(aspect_ratio)
     
     print("Generating for prompt:", prompt)
+    
     image = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
         width=width,
         height=height,
-        num_inference_steps=50,
-        true_cfg_scale=4.0,
-        generator=torch.Generator(device="cuda").manual_seed(42)
+        num_inference_steps=num_inference_steps,
+        true_cfg_scale=guidance_scale,
+        generator=torch.Generator(device="cuda").manual_seed(seed)
     ).images[0]
 
     #image.save("example.png")
@@ -74,8 +75,6 @@ def infer(
     return image, seed
 
 
-
-
 css = """
 #col-container {
     margin: 0 auto;
@@ -83,7 +82,6 @@ css = """
 }
 """
 
-
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         # gr.Markdown('<div style="text-align: center;"><a href="https://huggingface.co/Qwen/Qwen-Image"><img src="https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-Image/qwen_image_logo.png" width="400"/></a></div>')
@@ -102,26 +100,30 @@ with gr.Blocks(css=css) as demo:
         result = gr.Image(label="Result", show_label=False)
 
         with gr.Accordion("Advanced Settings", open=False):
-            negative_prompt = gr.Text(
-                label="Negative prompt",
-                max_lines=1,
-                placeholder="Enter a negative prompt",
-                visible=True,
-            )
 
-            seed = gr.Slider(
-                label="Seed",
-                minimum=0,
-                maximum=MAX_SEED,
-                step=1,
-                value=0,
-            )
+            with gr.Row():
+                negative_prompt = gr.Text(
+                    label="Negative prompt",
+                    max_lines=1,
+                    placeholder="Enter a negative prompt",
+                    visible=True,
+                )
+
+            with gr.Row():
+            
+                seed = gr.Slider(
+                    label="Seed",
+                    minimum=0,
+                    maximum=MAX_SEED,
+                    step=1,
+                    value=0,
+                )
 
-            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+                randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
             with gr.Row():
                 aspect_ratio = gr.Radio(
-                    label="Image size (ratio, max dim 1920)",
+                    label="Image size (ratio)",
                     choices=["1:1", "16:9", "9:16", "4:3", "3:4"],
                     value="16:9",
                 )