Add default weight configuration for Stage 1 and Stage 2 LoRAs in app.py, and update README.md to document new environment variables for customization.

README.md

@@ -43,10 +43,12 @@ The application uses environment variables for customization:
 ### LoRA Settings
 - `LIGHTNING_LORA_REPO`: Lightning 4-step LoRA repository (default: `lightx2v/Qwen-Image-Edit-2511-Lightning`)
 - `LIGHTNING_LORA_WEIGHT`: Lightning LoRA weight filename (default: `Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors`)
-- `STAGE1_LORA_REPO`: Repository for Stage 1 LoRA (anime → base body)
+- `STAGE1_LORA_REPO`: Repository for Stage 1 LoRA
 - `STAGE1_LORA_WEIGHT`: Weight filename for Stage 1 LoRA
-- `STAGE2_LORA_REPO`: Repository for Stage 2 LoRA (base body → guide body)
+- `STAGE1_WEIGHT_DEFAULT`: Default weight value for Stage 1 LoRA (default: `1.0`)
+- `STAGE2_LORA_REPO`: Repository for Stage 2 LoRA
 - `STAGE2_LORA_WEIGHT`: Weight filename for Stage 2 LoRA
+- `STAGE2_WEIGHT_DEFAULT`: Default weight value for Stage 2 LoRA (default: `1.0`)
 
 ### Prompt Settings
 - `STAGE1_PROMPT`: Prompt used for inference (default: "Convert anime character to base body structure")

app.py

@@ -22,6 +22,8 @@ STAGE1_LORA_REPO = os.environ.get("STAGE1_LORA_REPO", "default/stage1-lora")
 STAGE1_LORA_WEIGHT = os.environ.get("STAGE1_LORA_WEIGHT", "stage1.safetensors")
 STAGE2_LORA_REPO = os.environ.get("STAGE2_LORA_REPO", "default/stage2-lora")
 STAGE2_LORA_WEIGHT = os.environ.get("STAGE2_LORA_WEIGHT", "stage2.safetensors")
+STAGE1_WEIGHT_DEFAULT = float(os.environ.get("STAGE1_WEIGHT_DEFAULT", "1.0"))
+STAGE2_WEIGHT_DEFAULT = float(os.environ.get("STAGE2_WEIGHT_DEFAULT", "1.0"))
 STAGE1_PROMPT = os.environ.get("STAGE1_PROMPT", "Convert anime character to base body structure")
 STAGE2_PROMPT = os.environ.get("STAGE2_PROMPT", "Convert base body to clear guide body with structure lines")
 
@@ -180,12 +182,6 @@ with gr.Blocks(css=css) as demo:
             <h3 style="color: #5b47d1;">Anime Character Converter with Combined LoRAs</h3>
         </div>
         """)
-        gr.Markdown("""
-        Single-pass conversion powered by [Qwen-Image-Edit-2511](https://huggingface.co/Qwen/Qwen-Image-Edit-2511) with combined custom LoRAs.
-
-        **LoRAs Applied:** Lightning (4-step) + Stage1 + Stage2 simultaneously
-        """)
-
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("### 📥 Input")
@@ -219,14 +215,14 @@ with gr.Blocks(css=css) as demo:
                     minimum=0.0,
                     maximum=2.0,
                     step=0.1,
-                    value=1.0
+                    value=STAGE1_WEIGHT_DEFAULT
                 )
                 stage2_weight = gr.Slider(
                     label="Stage2 LoRA Weight",
                     minimum=0.0,
                     maximum=2.0,
                     step=0.1,
-                    value=1.0
+                    value=STAGE2_WEIGHT_DEFAULT
                 )
 
             gr.Markdown("### Generation Settings")