install yue2 wheel at runtime with --no-deps (build-time pins conflict with gradio6)

app.py

@@ -25,9 +25,34 @@ FEED_TTL = 30  # seconds
 pipe = None
 
 
+def _ensure_yue2():
+    """Install the vendored yue2-infer wheel at runtime (not build time).
+
+    It can't go in requirements.txt: it pins huggingface-hub==0.36.2 /
+    transformers==4.57.6 which conflict with Gradio 6 (hub>=1.16) and break
+    the Docker build. --no-deps sidesteps the pins; the package only needs
+    torch/transformers/hub/safetensors/tiktoken/soundfile which are present.
+    """
+    try:
+        import yue2  # noqa: F401
+        return
+    except ImportError:
+        pass
+    import subprocess
+    import sys
+    whl = os.path.join(os.path.dirname(os.path.abspath(__file__)), "yue2_infer-0.1.5-py3-none-any.whl")
+    print(f"[yue2] installing vendored wheel: {whl}")
+    subprocess.run(
+        [sys.executable, "-m", "pip", "install", "--no-deps", whl],
+        check=True, capture_output=True, text=True,
+    )
+    print("[yue2] installed.")
+
+
 def get_pipeline():
     global pipe
     if pipe is None:
+        _ensure_yue2()
         from yue2 import YuE2Pipeline
         print("Loading YuE2-3B pipeline on GPU...")
         pipe = YuE2Pipeline.from_pretrained(

requirements.txt

@@ -5,4 +5,4 @@ transformers>=4.40.0
 gradio
 huggingface-hub
 soundfile
-./yue2_infer-0.1.5-py3-none-any.whl
+tiktoken