Update app.py

app.py

@@ -10,9 +10,10 @@ from PIL import PngImagePlugin
 BASE_MODEL_ID = "stabilityai/stable-diffusion-xl-base-1.0"
 
 # LoRA réalisme global
-LORA_PHOTO_REPO = "ostris/photorealistic-slider-sdxl-lora"
-#LORA_PHOTO_REPO = "Muapi/hourglass-body-shape-sd1.5-sdxl-pony-flux-olaz"
-LORA_PHOTO_ADAPTER = "photo_realism"
+#LORA_PHOTO_REPO = "ostris/photorealistic-slider-sdxl-lora"
+LORA_PHOTO_REPO = "Muapi/hourglass-body-shape-sd1.5-sdxl-pony-flux-olaz"
+#LORA_PHOTO_ADAPTER = "photo_realism"
+LORA_PHOTO_ADAPTER = "hourglass_body"
 
 # LoRA amélioration de visages
 #LORA_FACE_REPO = "ostris/face-helper-sdxl-lora"
@@ -31,12 +32,19 @@ pipe = AutoPipelineForText2Image.from_pretrained(
 pipe.to(DEVICE)
 pipe.set_progress_bar_config(disable=True)
 
-print("Chargement LoRA photorealistic...")
+#print("Chargement LoRA photorealistic...")
+#pipe.load_lora_weights(
+#    LORA_PHOTO_REPO,
+#    adapter_name=LORA_PHOTO_ADAPTER,
+#)
+print("Chargement LoRA hourglass body...")
 pipe.load_lora_weights(
-    LORA_PHOTO_REPO,
-    adapter_name=LORA_PHOTO_ADAPTER,
+    LORA_HOURGLASS_FILE,  # chemin local relatif
+    weight_name=LORA_HOURGLASS_FILE,
+    adapter_name=LORA_HOURGLASS_ADAPTER,
 )
 
+
 print("Chargement LoRA face-helper...")
 pipe.load_lora_weights(
     LORA_FACE_REPO,
@@ -92,6 +100,12 @@ def generate(
         adapters.append(LORA_FACE_ADAPTER)
         weights.append(0.0)
 
+    # NOUVEAU : Hourglass body
+    hourglass_enabled = True  # ou passe en param UI
+    hourglass_weight = 0.9   # recommandé par l'auteur [web:19]
+    adapters.append(LORA_HOURGLASS_ADAPTER)
+    weights.append(float(hourglass_weight) if hourglass_enabled else 0.0)
+    
     pipe.set_adapters(adapters, adapter_weights=weights)
 
     try:
@@ -214,6 +228,7 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
                 label="Force LoRA photorealistic",
             )
 
+            
             face_enabled = gr.Checkbox(
                 label="Activer LoRA face-helper (visages)",
                 value=True,
@@ -231,6 +246,19 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
                 value="sdxl_lora_generation",
             )
 
+            hourglass_enabled = gr.Checkbox(
+                label="Activer LoRA hourglass body shape",
+                value=True,
+            )
+
+            hourglass_weight = gr.Slider(
+                minimum=0.0,
+                maximum=2.0,
+                value=0.9,  # valeur recommandée [web:19]
+                step=0.05,
+                label="Force LoRA hourglass body",
+            )
+            
             run_btn = gr.Button("Générer", variant="primary")
 
         with gr.Column():
@@ -260,6 +288,8 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
             face_enabled,
             face_weight,
             script_name,
+            hourglass_enabled,
+            hourglass_weight,
         ],
         outputs=[out_img, out_script, out_file],
     )