Update app.py

app.py

@@ -49,14 +49,16 @@ pipe.set_adapters(["light_migration"], adapter_weights=[1.])
 pipe.fuse_lora(adapter_names=["light_migration"], lora_scale=1.25)
 pipe.unload_lora_weights()
 
-pipe.transformer.__class__ = QwenImageTransformer2DModel
-pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
+spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/Qwen-Image", variant="fa3")
 
-optimize_pipeline_(
-    pipe,
-    image=[Image.new("RGB", (1024, 1024)), Image.new("RGB", (1024, 1024))],
-    prompt="prompt"
-)
+#pipe.transformer.__class__ = QwenImageTransformer2DModel
+#pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
+
+#optimize_pipeline_(
+#    pipe,
+#    image=[Image.new("RGB", (1024, 1024)), Image.new("RGB", (1024, 1024))],
+#    prompt="prompt"
+#)
 
 MAX_SEED = np.iinfo(np.int32).max
 
@@ -194,7 +196,7 @@ css = '''
 .image-container { min-height: 300px; }
 '''
 
-with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
+with gr.Blocks() as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("## 💡 Qwen Image Edit — Light Migration")
         gr.Markdown("""
@@ -299,4 +301,4 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     # API endpoint
     gr.api(infer_light_migration, api_name="infer_light_migration")
 
-demo.launch(mcp_server=True)
+demo.launch(mcp_server=True, theme=gr.themes.Citrus(), css=css, footer_links=["api", "gradio", "settings"])