Update app.py

app.py

@@ -16,11 +16,11 @@ from diffusers import (
     PNDMScheduler,
 )
 
-# ----------------- Config (set in Space Secrets if private) -----------------
+# Config (set in Space Secrets if private)
 MODEL_REPO_ID = os.getenv("MODEL_REPO_ID", "DB2169/CyberPony_Lora").strip()
 CHECKPOINT_FILENAME = os.getenv("CHECKPOINT_FILENAME", "SAFETENSORS_FILENAME.safetensors").strip()
 HF_TOKEN = os.getenv("HF_TOKEN", None)
-DO_WARMUP = os.getenv("WARMUP", "1") == "1"   # set WARMUP=0 to skip the first warmup call
+DO_WARMUP = os.getenv("WARMUP", "1") == "1"  # set WARMUP=0 to skip the first warmup call
 
 # Optional override: JSON string for LoRA manifest (same shape as loras.json)
 LORAS_JSON = os.getenv("LORAS_JSON", "").strip()
@@ -44,10 +44,8 @@ IS_SDXL = True
 LORA_MANIFEST: Dict[str, Dict[str, str]] = {}
 INIT_ERROR: Optional[str] = None
 
-# ----------------- Helpers -----------------
 def load_lora_manifest(repo_dir: str) -> Dict[str, Dict[str, str]]:
-    """
-    Manifest load order:
+    """Manifest load order:
     1) Environment variable LORAS_JSON (if provided)
     2) loras.json inside the downloaded model repo
     3) loras.json at the Space root (next to app.py)
@@ -93,7 +91,6 @@ def load_lora_manifest(repo_dir: str) -> Dict[str, Dict[str, str]]:
         }
     }
 
-# ----------------- Bootstrap (download + load on CPU) -----------------
 def bootstrap_model():
     """
     Downloads MODEL_REPO_ID into REPO_DIR and loads the single-file checkpoint,
@@ -179,7 +176,6 @@ def apply_loras(selected: List[str], scale: float, repo_dir: str):
     except Exception as e:
         print(f"[WARN] set_adapters failed: {e}")
 
-# ----------------- Generation (ZeroGPU) -----------------
 @spaces.GPU
 def txt2img(
     prompt: str,
@@ -236,7 +232,7 @@ def warmup():
     except Exception as e:
         print(f"[WARN] Warmup failed: {e}")
 
-# ----------------- UI -----------------
+# UI
 with gr.Blocks(title="SDXL Space (ZeroGPU, single-file, LoRA-ready)") as demo:
     status = gr.Markdown("")
 
@@ -286,4 +282,4 @@ with gr.Blocks(title="SDXL Space (ZeroGPU, single-file, LoRA-ready)") as demo:
         concurrency_id="gpu_queue",
     )
 
-demo.queue(max_size=32, default_concurrency_limit=1).launch()
+    demo.queue(max_size=32, default_concurrency_limit=1).launch()