Update app.py

app.py

@@ -1,18 +1,18 @@
-# app.py — InstantID × Beautiful Realistic Asians v7（ZeroGPU 対応フル版）
+# app.py — InstantID × Beautiful Realistic Asians v7（ZeroGPU 対応・CrucibleAI ControlNet）
 # 2025-06-22
 
 ##############################################################################
-# 0. 旧 API → 新 API 互換パッチ（必ず diffusers より前に実行）
+# 0. 旧 API → 新 API 互換パッチ（必ず diffusers import の前に置く）
 ##############################################################################
 from huggingface_hub import hf_hub_download
 import huggingface_hub as _hf_hub
 
-# diffusers-0.27 が import する cached_download() を v0.28+ でも使えるように別名定義
+# diffusers-0.27 が参照する cached_download() を v0.28+ でも使えるように注入
 if not hasattr(_hf_hub, "cached_download"):
     _hf_hub.cached_download = hf_hub_download
 
 ##############################################################################
-# 1. 標準ライブラリ & 外部ライブラリ
+# 1. 標準 & 外部ライブラリ
 ##############################################################################
 import os, io, base64, subprocess, traceback
 from pathlib import Path
@@ -37,7 +37,7 @@ from basicsr.utils.download_util import load_file_from_url
 from realesrgan import RealESRGANer
 
 ##############################################################################
-# 2. キャッシュ／永続ディレクトリ
+# 2. キャッシュ & パス
 ##############################################################################
 PERSIST_BASE = Path("/data")
 CACHE_ROOT = (
@@ -72,7 +72,7 @@ REALESRGAN_URL = (
 )
 
 ##############################################################################
-# 4. ユーティリティ：堅牢ダウンロード
+# 4. ダウンローダ
 ##############################################################################
 def download(url: str, dst: Path, attempts: int = 2):
     if dst.exists():
@@ -87,7 +87,7 @@ def download(url: str, dst: Path, attempts: int = 2):
     return dst
 
 ##############################################################################
-# 5. グローバル（lazy-load される）
+# 5. グローバル変数（lazy-load）
 ##############################################################################
 pipe: Optional[StableDiffusionControlNetPipeline] = None
 face_analyser: Optional[FaceAnalysis] = None
@@ -97,26 +97,26 @@ upsampler: Optional[RealESRGANer] = None
 # 6. パイプライン初期化
 ##############################################################################
 def initialize_pipelines():
-    """最初の推論時に一度だけ呼び出す"""
     global pipe, face_analyser, upsampler
     if pipe is not None:
         return
 
     print("[INIT] Downloading model assets …")
 
-    # (1) ベースモデル & IP-Adapter
+    # 6-1 ベースモデル & IP-Adapter
     bra_ckpt = download(BRA_V7_URL, MODELS_DIR / "bra_v7.safetensors")
     ip_bin = download(IP_ADAPTER_BIN_URL, MODELS_DIR / "ip_adapter.bin")
     ip_lora = download(IP_ADAPTER_LORA_URL, LORA_DIR / "ip_adapter_faceid.lora")
 
-    # (2) ControlNet (InstantID)
+    # 6-2 ControlNet（CrucibleAI / diffusion_sd15）
     controlnet = ControlNetModel.from_pretrained(
-        "InstantID/ControlNet-Mediapipe-Face",
+        "CrucibleAI/ControlNetMediaPipeFace",  # ← 公開リポジトリ :contentReference[oaicite:0]{index=0}
+        subfolder="diffusion_sd15",            # ← SD-1.5 用フォルダ :contentReference[oaicite:1]{index=1}
         torch_dtype=torch.float16,
         cache_dir=str(MODELS_DIR),
     )
 
-    # (3) Diffusers Pipeline
+    # 6-3 Diffusers パイプライン
     pipe_tmp = StableDiffusionControlNetPipeline.from_pretrained(
         "runwayml/stable-diffusion-v1-5",
         controlnet=controlnet,
@@ -132,10 +132,9 @@ def initialize_pipelines():
         subfolder="scheduler",
         cache_dir=str(MODELS_DIR),
     )
-    # IP-Adapter：LoRA を適用
+    # IP-Adapter
     pipe_tmp.load_ip_adapter(ip_bin)
-    ip_layers = AttnProcsLayers(pipe_tmp.unet.attn_processors)
-    ip_layers.load_lora_weights(
+    AttnProcsLayers(pipe_tmp.unet.attn_processors).load_lora_weights(
         ip_lora, adapter_name="ip_faceid", safe_load=True
     )
     pipe_tmp.set_adapters(["ip_faceid"], adapter_weights=[0.6])
@@ -143,13 +142,13 @@ def initialize_pipelines():
 
     pipe = pipe_tmp
 
-    # (4) InsightFace
+    # 6-4 InsightFace
     face_analyser = FaceAnalysis(
         name="buffalo_l", root=str(MODELS_DIR), providers=["CUDAExecutionProvider"]
     )
     face_analyser.prepare(ctx_id=0, det_size=(640, 640))
 
-    # (5) Real-ESRGAN
+    # 6-5 Real-ESRGAN
     esrgan_ckpt = download(REALESRGAN_URL, UPSCALE_DIR / "realesrgan_x4plus.pth")
     upsampler = RealESRGANer(
         scale=4,
@@ -177,7 +176,7 @@ NEG_PROMPT = (
 )
 
 ##############################################################################
-# 8. 生成関数（GPU を掴む）
+# 8. 生成関数（GPU アタッチ）
 ##############################################################################
 @spaces.GPU(duration=60)
 def generate_core(
@@ -198,10 +197,9 @@ def generate_core(
         if pipe is None:
             initialize_pipelines()
 
-        np_face = np.array(face_img)
-        faces = face_analyser.get(np_face)
+        faces = face_analyser.get(np.array(face_img))
         if len(faces) == 0:
-            raise ValueError("顔が検出できませんでした。別の画像でお試しください。")
+            raise ValueError("顔が検出できませんでした。別の画像をお試しください。")
 
         pipe.set_adapters(["ip_faceid"], adapter_weights=[ip_scale])
 
@@ -275,7 +273,7 @@ with gr.Blocks(title="InstantID × BRA v7 (ZeroGPU)") as demo:
     )
 
 ##############################################################################
-# 10. FastAPI エンドポイント
+# 10. FastAPI REST
 ##############################################################################
 app = FastAPI()
 
@@ -314,6 +312,6 @@ async def api_generate(
         raise HTTPException(status_code=500, detail=str(e))
 
 ##############################################################################
-# 11. Launch（Gradio が自動で Uvicorn を立ち上げる）
+# 11. Launch（Gradio が自動で Uvicorn を起動）
 ##############################################################################
 demo.queue(default_concurrency_limit=2).launch(share=False)