Update app.py

app.py

@@ -56,8 +56,6 @@ 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}")
@@ -72,10 +70,10 @@ def load_lora_pipeline(revision: str):
     pipe.load_lora_weights(lora_path)
     return pipe
 
-def init_pipelines(style_intensity: str = "low"):
+def init_pipelines(style_intensity: str = "medium"):
     global pipeline_base, pipeline_lora
     revision_map = {
-        "low": "step_750",
+        "low": "step_0750",
         "medium": "step_1500",
         "high": "step_2500"
     }
@@ -90,7 +88,7 @@ def init_pipelines(style_intensity: str = "low"):
     logger.info("LoRA pipeline loaded")
 
 # Initial pipelines
-init_pipelines("low")
+init_pipelines("medium")
 
 def update_lora_pipeline(style_intensity):
     logger.info(f"Updating LoRA pipeline to style: {style_intensity}")
@@ -140,7 +138,7 @@ with gr.Blocks() as demo:
             style_intensity = gr.Dropdown(
                 label="LoRA Style Intensity",
                 choices=["low", "medium", "high"],
-                value="low",
+                value="medium",
                 interactive=True
             )