Update requirements.txt

requirements.txt

@@ -1,13 +1,40 @@
---extra-index-url https://download.pytorch.org/whl/cu113
-torch
-#https://storage.googleapis.com/libtpu-nightly-releases/wheels/libtpu-nightly/libtpu_nightly-0.1.dev20241010+nightly.cleanup-py3-none-any.whl
-#torch_xla[tpu] == 2.6.0
-torchaudio
+# ---------------------------------------------------------------------------
+# Sonogram app -- requirements.txt for the Hugging Face Space
+# Replaces the original, which was missing scikit-learn entirely and
+# pinned an outdated CUDA 11.3 wheel index that isn't needed on HF's
+# managed hardware.
+#
+# NOTE: unlike a self-hosted server, HF Spaces provisions the CUDA runtime
+# for you based on the Space's selected hardware tier (CPU / T4 / A10G /
+# etc.), so there's no need for a custom --extra-index-url here -- plain
+# `torch`/`torchaudio` from PyPI will pull a CUDA-enabled build automatically
+# when the Space's hardware tier has a GPU.
+# ---------------------------------------------------------------------------
+
+torch>=2.5.1
+torchaudio>=2.5.1
+
+# --- Core diarization pipeline ------------------------------------------
+pyannote.audio==3.4.0
+numpy<2.0          # pyannote's older internals hit a removed np.NaN alias on
+                    # numpy 2.x; this is the combination verified working
+                    # with pyannote.audio 3.4.0.
 soundfile
-numpy
+matplotlib          # imported internally by pyannote.audio's task modules,
+                    # even though this app doesn't plot with it directly.
+
+# huggingface_hub is intentionally left unpinned: pip resolves a compatible
+# version automatically as a dependency of pyannote.audio==3.4.0.
+
+# --- Classifier support --------------------------------------------------
+# Pin to match (or be very close to) the version 05062026_groupClassifier.pkl
+# was actually trained with, to avoid unpickling warnings/version drift.
+scikit-learn==1.7.2
+
+# --- App / UI -------------------------------------------------------------
 streamlit == 1.45.1
-opencv-python
-pyannote.audio
-matplotlib
+pandas
+opencv-python-headless   # headless variant: no GUI/X11 needed on Spaces;
+                          # lighter and avoids extra system-level dependencies.
 plotly
 kaleido==1.0.0