Update app.py

app.py

@@ -21,7 +21,9 @@ from insightface.app import FaceAnalysis
 # Config globale
 # ---------------------------
 
-BASE_MODEL_ID = "stabilityai/stable-diffusion-xl-base-1.0"
+# ⚠️ Mets ici TON repo diffusers ArtFusion (après conversion + push_to_hub)
+BASE_MODEL_ID = "leteint/artfusion-xl-real-v16-diffusers"
+
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
 
@@ -33,20 +35,26 @@ CHECKPOINT_DIR = "./checkpoints"
 # ---------------------------
 
 def download_checkpoints():
+    # ControlNet InstantID (dossier ControlNetModel complet)
     hf_hub_download(
         repo_id="InstantX/InstantID",
         filename="ControlNetModel/config.json",
         local_dir=CHECKPOINT_DIR,
+        local_dir_use_symlinks=False,
     )
     hf_hub_download(
         repo_id="InstantX/InstantID",
         filename="ControlNetModel/diffusion_pytorch_model.safetensors",
         local_dir=CHECKPOINT_DIR,
+        local_dir_use_symlinks=False,
     )
+
+    # IP-Adapter InstantID
     hf_hub_download(
         repo_id="InstantX/InstantID",
         filename="ip-adapter.bin",
         local_dir=CHECKPOINT_DIR,
+        local_dir_use_symlinks=False,
     )
 
 
@@ -86,13 +94,14 @@ def get_face_info(image: Image.Image):
         reverse=True,
     )[0]
 
-    # Selon la version d'insightface, accès par attribut ou dict
+    # Embedding
     emb = getattr(face, "normed_embedding", None)
     if emb is None and isinstance(face, dict):
         emb = face.get("embedding", None)
     if emb is None:
         raise RuntimeError("Impossible de récupérer l'embedding du visage.")
 
+    # Keypoints
     kps = getattr(face, "kps", None)
     if kps is None and isinstance(face, dict):
         kps = face.get("kps", None)
@@ -103,15 +112,17 @@ def get_face_info(image: Image.Image):
 
 
 # ---------------------------
-# Chargement du pipeline InstantID SDXL
+# Chargement du pipeline InstantID SDXL + ArtFusion
 # ---------------------------
 
 def load_pipeline():
+    # ControlNet InstantID
     controlnet = ControlNetModel.from_pretrained(
         CONTROLNET_PATH,
         torch_dtype=DTYPE,
     )
 
+    # Pipeline InstantID SDXL avec ArtFusion comme base
     pipe = StableDiffusionXLInstantIDPipeline.from_pretrained(
         BASE_MODEL_ID,
         controlnet=controlnet,
@@ -120,7 +131,10 @@ def load_pipeline():
 
     if DEVICE == "cuda":
         pipe.to("cuda")
+    else:
+        pipe.to("cpu")
 
+    # IP-Adapter InstantID
     pipe.load_ip_adapter_instantid(IP_ADAPTER_PATH)
     pipe.set_ip_adapter_scale(0.6)
 
@@ -145,7 +159,7 @@ def generate(face_image, prompt, negative_prompt="", steps=30, guidance_scale=5,
         if face_image is None:
             raise gr.Error("Merci de fournir une image de visage.")
 
-        # Device
+        # On s'assure que le pipe est bien sur le bon device
         if DEVICE == "cuda":
             pipe.to("cuda")
         else:
@@ -159,7 +173,6 @@ def generate(face_image, prompt, negative_prompt="", steps=30, guidance_scale=5,
         kps_image = draw_kps(face_image, face_kps)       # PIL.Image, comme dans l'exemple officiel
 
         # 3) Appel du pipeline InstantID SDXL
-        # Note : image_embeds = face_emb, image = kps_image (comme dans l'exemple)
         out = pipe(
             prompt=prompt,
             negative_prompt=negative_prompt,
@@ -185,7 +198,7 @@ def generate(face_image, prompt, negative_prompt="", steps=30, guidance_scale=5,
 # ---------------------------
 
 with gr.Blocks() as demo:
-    gr.Markdown("## InstantID + SDXL (ZeroGPU) – 1 image visage → corps généré")
+    gr.Markdown("## InstantID + ArtFusion XL Real (SDXL) – Visage → corps généré (ZeroGPU)")
 
     with gr.Row():
         with gr.Column():
@@ -195,8 +208,9 @@ with gr.Blocks() as demo:
                 label="Prompt",
                 lines=3,
                 value=(
-                    "photorealistic full body portrait, firefighter uniform with detailed fabric, "
-                    "realistic studio lighting, 35mm DSLR, sharp focus on face, clean background, high resolution"
+                    "photorealistic full body portrait, white European man, realistic skin texture, "
+                    "firefighter uniform with detailed fabric, realistic studio lighting, 35mm DSLR, "
+                    "sharp focus on face, clean background, high resolution"
                 ),
             )