Update app.py

app.py

@@ -56,6 +56,8 @@ def load_lora_pipeline(revision: str):
 
     # Extract step number from revision name (e.g., "step_1500" -> "1500")
     step_num = revision.split('_')[-1]
+    if len(step_num) < 4:
+        step_num = '0' + step_num
     filename = f"moroccan_ghibli_flux_lora_00000{step_num}.safetensors"
 
     logger.info(f"Downloading LoRA weights: {filename}")
@@ -73,7 +75,7 @@ def load_lora_pipeline(revision: str):
 def init_pipelines(style_intensity: str = "medium"):
     global pipeline_base, pipeline_lora
     revision_map = {
-        "low": "step_0750",
+        "low": "step_750",
         "medium": "step_1500",
         "high": "step_2500"
     }