[Admin maintenance] Support new ZeroGPU hardware

app.py

@@ -1,3 +1,24 @@
+# The `solospeech` package (installed from git in requirements.txt) hard-pins
+# torch==2.4.1/torchaudio==2.4.1/torchvision==0.19.1 in its own setup.py.
+# Those wheels predate NVIDIA Blackwell (sm_120) and have no compatible CUDA
+# kernels, causing `RuntimeError: CUDA error: no kernel image is available
+# for execution on the device` on ZeroGPU's current hardware. We can't loosen
+# solospeech's pin from here (it's an upstream dependency, and requirements.txt
+# is resolved in a single pip invocation where an explicit newer torch pin
+# conflicts with solospeech's exact pin). So: let the build install whatever
+# solospeech demands, then upgrade torch to a Blackwell-compatible build at
+# container startup, before torch (or anything importing it) is loaded.
+import subprocess
+import sys
+
+subprocess.run(
+    [
+        sys.executable, "-m", "pip", "install", "--no-cache-dir",
+        "torch==2.8.0", "torchaudio==2.8.0", "torchvision==0.23.0",
+    ],
+    check=True,
+)
+
 import gradio as gr
 import spaces
 import yaml
@@ -9,6 +30,36 @@ import torch.nn.functional as F
 import librosa
 from tqdm import tqdm
 from diffusers import DDIMScheduler
+
+# speechbrain==1.0.2 (pinned transitively by the `solospeech` package) still
+# calls huggingface_hub.hf_hub_download(..., use_auth_token=...). Newer
+# huggingface_hub releases dropped that kwarg in favor of `token`, which
+# raises `TypeError: hf_hub_download() got an unexpected keyword argument
+# 'use_auth_token'`. Shim it here so old callers keep working.
+import huggingface_hub
+
+_orig_hf_hub_download = huggingface_hub.hf_hub_download
+
+
+def _hf_hub_download_compat(*args, **kwargs):
+    if "use_auth_token" in kwargs:
+        kwargs["token"] = kwargs.pop("use_auth_token")
+    return _orig_hf_hub_download(*args, **kwargs)
+
+
+huggingface_hub.hf_hub_download = _hf_hub_download_compat
+
+# speechbrain's fetch() catches `requests.exceptions.HTTPError` to gracefully
+# skip an optional file (e.g. the default "custom.py") that doesn't exist in
+# a model repo. Newer huggingface_hub raises `RemoteEntryNotFoundError`
+# (built on httpx, not requests) for the same 404, so that except clause no
+# longer matches and the error propagates instead of being swallowed. Widen
+# the caught exception type so the old graceful-skip behavior still works.
+import speechbrain.utils.fetching as _sb_fetching
+from huggingface_hub.errors import HfHubHTTPError as _HfHubHTTPError
+
+_sb_fetching.HTTPError = (_sb_fetching.HTTPError, _HfHubHTTPError)
+
 from solospeech.model.solospeech.conditioners import SoloSpeech_TSE
 # from solospeech.model.solospeech.conditioners import SoloSpeech_TSR
 from solospeech.scripts.solospeech.utils import save_audio