Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-os.system('pip install --upgrade spaces')
 
 import gradio as gr
 import numpy as np
@@ -176,12 +175,21 @@ pipe = QwenImageEditPipelineCustom.from_pretrained(
 
 # Load Lightning LoRA weights for acceleration
 try:
+    pipe.load_lora_weights(
+        "eigen-ai-labs/eigen-banana-qwen-image-edit", 
+        weight_name="eigen-banana-qwen-image-edit-fp16-lora.safetensors",
+        adapter_name="eigen"
+    )
+    
     pipe.load_lora_weights(
         "lightx2v/Qwen-Image-Lightning", 
-        weight_name="Qwen-Image-Lightning-8steps-V1.1.safetensors"
+        weight_name="Qwen-Image-Lightning-4steps-V2.0-bf16.safetensors",
+        adapter_name="lightning"
     )
-    pipe.fuse_lora()
-    print("Successfully loaded Lightning LoRA weights")
+    pipe.set_adapters(["eigen", "lightning"], adapter_weights=[1., 1.])
+    pipe.fuse_lora(adapter_names=["eigen", "lightning"], lora_scale=1)
+    pipe.unload_lora_weights()
+    print("Successfully loaded LoRA weights")
 except Exception as e:
     print(f"Warning: Could not load Lightning LoRA weights: {e}")
     print("Continuing with base model...")
@@ -255,7 +263,7 @@ examples = [
 ]
 
 css = """
-#col-container {
+#col-container, #examples {
     margin: 0 auto;
     max-width: 1024px;
 }
@@ -266,20 +274,14 @@ css = """
     width: 400px;
 }
 #edit_text{margin-top: -62px !important}
+.dark .progress-text{color: white !important}
 """
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.HTML("""
-        <div id="logo-title">
-            <img src="https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-Image/qwen_image_edit_logo.png" alt="Qwen-Image Edit Logo" width="400" style="display: block; margin: 0 auto;">
-            <h2 style="font-style: italic;color: #5b47d1;margin-top: -27px !important;margin-left: 96px">Fast, 8-steps with Lightning LoRA</h2>
-        </div>
-        """)
-        gr.Markdown("""
-        [Learn more](https://github.com/QwenLM/Qwen-Image) about the Qwen-Image series. 
-        This demo uses the [Qwen-Image-Lightning](https://huggingface.co/lightx2v/Qwen-Image-Lightning) LoRA for accelerated inference.
-        Try on [Qwen Chat](https://chat.qwen.ai/), or [download model](https://huggingface.co/Qwen/Qwen-Image-Edit) to run locally with ComfyUI or diffusers.
+        gr.Markdown("""# Eigen Banana""")
+        gr.Markdown(""" [Eigen-Banana-Qwen-Image-Edit LoRA](https://huggingface.co/eigen-ai-labs/eigen-banana-qwen-image-edit) trained on [Apple's Pico-Banana-400k dataset](https://github.com/apple/pico-banana-400k) 
+            with [lightx2v/Qwen-Image-Lightning](https://huggingface.co/lightx2v/Qwen-Image-Lightning) for 4-step inference 💨
         """)
         
         with gr.Row():
@@ -330,7 +332,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=4,
                     maximum=28,
                     step=1,
-                    value=8
+                    value=4
                 )
                 
             # Removed num_images_per_prompt slider entirely