Update app.py and README.md to enhance environment variable configuration for model and LoRA settings, improving customization options.

README.md

@@ -37,15 +37,20 @@ Built on [Qwen-Image-Edit-2511](https://huggingface.co/Qwen/Qwen-Image-Edit-2511
 
 The application uses environment variables for customization:
 
+### Model Settings
+- `BASE_MODEL`: Base model repository (default: `Qwen/Qwen-Image-Edit-2511`)
+
 ### 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_WEIGHT`: Weight filename for Stage 1 LoRA
 - `STAGE2_LORA_REPO`: Repository for Stage 2 LoRA (base body → guide body)
 - `STAGE2_LORA_WEIGHT`: Weight filename for Stage 2 LoRA
 
 ### Prompt Settings
-- `STAGE1_PROMPT`: Prompt for Stage 1 conversion
-- `STAGE2_PROMPT`: Prompt for Stage 2 conversion
+- `STAGE1_PROMPT`: Prompt for Stage 1 conversion (default: "Convert anime character to base body structure")
+- `STAGE2_PROMPT`: Prompt for Stage 2 conversion (default: "Convert base body to clear guide body with structure lines")
 
 ## Usage
 

app.py

@@ -14,7 +14,10 @@ from diffusers import FlowMatchEulerDiscreteScheduler, QwenImageEditPlusPipeline
 import math
 import os
 
-# --- Environment Variables for LoRA and Prompts ---
+# --- Environment Variables for Model, LoRA and Prompts ---
+BASE_MODEL = os.environ.get("BASE_MODEL", "Qwen/Qwen-Image-Edit-2511")
+LIGHTNING_LORA_REPO = os.environ.get("LIGHTNING_LORA_REPO", "lightx2v/Qwen-Image-Edit-2511-Lightning")
+LIGHTNING_LORA_WEIGHT = os.environ.get("LIGHTNING_LORA_WEIGHT", "Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors")
 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")
@@ -48,14 +51,14 @@ scheduler_config = {
 scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
 
 # Load single shared pipeline
-pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2511",
+pipe = QwenImageEditPlusPipeline.from_pretrained(BASE_MODEL,
                                                  scheduler=scheduler,
                                                  torch_dtype=dtype).to(device)
 # Load all LoRAs but don't fuse yet
 # Load 4-step Lightning LoRA
 pipe.load_lora_weights(
-        "lightx2v/Qwen-Image-Edit-2511-Lightning",
-        weight_name="Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors",
+        LIGHTNING_LORA_REPO,
+        weight_name=LIGHTNING_LORA_WEIGHT,
         adapter_name="lightning"
 )
 # Load Stage 1 LoRA
@@ -195,8 +198,6 @@ with gr.Blocks(css=css) as demo:
         </div>
         """)
         gr.Markdown("""
-        Two-stage conversion pipeline powered by [Qwen-Image-Edit-2511](https://huggingface.co/Qwen/Qwen-Image-Edit-2511) with custom LoRAs.
-
         **Stage 1:** Converts anime characters to base body structure
         **Stage 2:** Converts base body to clear guide body with structure lines
         """)