Update Dockerfile

Dockerfile

@@ -57,38 +57,35 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 RUN ln -sf /usr/bin/python3.11 /usr/bin/python && python -m pip install --upgrade pip
 
 # --- Python deps (pin order matters!) ---
-# 1) JAX CUDA pins
+# 1) JAX
 RUN python -m pip install "jax[cuda12]==0.7.1" "jaxlib==0.7.1"
 
-# 2) Lock seqio early to avoid backtracking madness
+# 2) seqio early
 RUN python -m pip install "seqio==0.0.11"
 
-# 3) Install Magenta RT *without* deps so we control pins
-RUN python -m pip install --no-deps 'git+https://github.com/magenta/magenta-realtime#egg=magenta_rt[gpu]'
-
-# 4) TF nightlies (MATCH DATES!)
-RUN python -m pip install \
-    "tf-nightly" \
-    "tensorflow-text-nightly" \
-    "tf-hub-nightly" \
-    "tf2jax"
+# 3) TF nightlies - MATCHING DATES!
+RUN python -m pip install --no-cache-dir \
+    "tf-nightly==2.20.0.dev20250619" \
+    "tensorflow-text-nightly==2.20.0.dev20250619" \
+    "tf-hub-nightly"
 
-RUN python -m pip install \
-    "tf2jax"
+# 4) magenta-rt without deps
+RUN python -m pip install --no-deps \
+    'git+https://github.com/magenta/magenta-realtime#egg=magenta_rt[gpu]'
 
-# 6) The rest of MRT deps + API runtime deps
+# 5) tf2jax + other deps
 RUN python -m pip install \
-    gin-config librosa resampy soundfile \
-    google-auth google-auth-oauthlib google-auth-httplib2 \
-    google-api-core googleapis-common-protos google-resumable-media \
-    google-cloud-storage requests tqdm typing-extensions numpy==2.1.3 \
-    fastapi uvicorn[standard] python-multipart pyloudnorm
+    tf2jax gin-config librosa resampy soundfile \
+    google-auth google-cloud-storage numpy==2.1.3 \
+    fastapi uvicorn[standard] python-multipart pyloudnorm \
+    huggingface_hub gradio soxr
 
-# 7) Exact commits for T5X/Flaxformer as in pyproject
+# 6) t5x/flaxformer last
 RUN python -m pip install \
     "t5x @ git+https://github.com/google-research/t5x.git@92c5b46" \
     "flaxformer @ git+https://github.com/google/flaxformer@399ea3a"
 
+
 # ---- FINAL: enforce TF nightlies and clean any stable TF ----
 RUN python - <<'PY'
 import sys, sysconfig, glob, os, shutil