Update app.py

app.py

@@ -152,37 +152,37 @@ MAX_SEED = np.iinfo(np.int32).max
 def get_image_size(aspect_ratio):
     """Converts aspect ratio string to width, height tuple."""
     if aspect_ratio == "1:1":
-        return 1328, 1328
+        return 664, 664
     elif aspect_ratio == "16:9":
         return 832, 464
     elif aspect_ratio == "9:16":
-        return 928, 1664
+        return 464, 832
     elif aspect_ratio == "4:3":
-        return 1472, 1104
+        return 736, 552
     elif aspect_ratio == "3:4":
-        return 1104, 1472
+        return 552, 736
     elif aspect_ratio == "3:2":
-        return 1584, 1056
+        return 792, 528
     elif aspect_ratio == "2:3":
-        return 1056, 1584
+        return 528, 792
     else:
         # Default to 1:1 if something goes wrong
-        return 1328, 1328
+        return 664, 664
 
 # --- Main Inference Function (with hardcoded negative prompt) ---
-@spaces.GPU(duration=120)
-def infer(
+@spaces.GPU(duration=30)
+def generate_image(
     prompt,
     seed=42,
     randomize_seed=False,
     aspect_ratio="16:9",
     guidance_scale=4.0,
-    num_inference_steps=50,
+    num_inference_steps=24,
 #    prompt_enhance=True,
     progress=gr.Progress(track_tqdm=True),
 ):
     """
-    Generates an image using the local Qwen-Image diffusers pipeline.
+    Generate an image using Qwen Image.
     """
     # Hardcode the negative prompt as requested
     negative_prompt = "text, watermark, copyright, blurry, low resolution"
@@ -290,11 +290,11 @@ with gr.Blocks(css=css) as demo:
                     value=24,
                 )
 
-        gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=infer, cache_examples=False)
+        gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=generate_image, cache_examples=False)
 
     gr.on(
         triggers=[run_button.click, prompt.submit],
-        fn=infer,
+        fn=generate_image,
         inputs=[
             prompt,
             # negative_prompt is no longer an input from the UI
@@ -309,4 +309,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(mcp_server=True)