Update Dockerfile

Dockerfile

@@ -57,35 +57,37 @@ 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
+# 1) Install JAX
 RUN python -m pip install "jax[cuda12]==0.7.1" "jaxlib==0.7.1"
 
-# 2) seqio early
+# 2) Pin protobuf BEFORE anything else
+RUN python -m pip install "protobuf>=4.21.0,<5.0.0"
+
+# 3) Install seqio
 RUN python -m pip install "seqio==0.0.11"
 
-# 3) TF nightlies - MATCHING DATES!
-RUN python -m pip install --no-cache-dir \
-    "tf-nightly==2.20.0.dev20250708" \
-    "tensorflow-text-nightly" \
-    "tf-hub-nightly"
+# 4) Install TF stable (should respect protobuf constraint)
+RUN python -m pip install \
+    "tensorflow==2.17.0" \
+    "tensorflow-text==2.17.0" \
+    "tensorflow-hub==0.16.1"
 
-# 4) magenta-rt without deps
+# 5) Install magenta-rt without deps
 RUN python -m pip install --no-deps \
     'git+https://github.com/magenta/magenta-realtime#egg=magenta_rt[gpu]'
 
-# 5) tf2jax + other deps
+# 6) Rest of deps
 RUN python -m pip install \
     tf2jax gin-config librosa resampy soundfile \
-    google-auth google-cloud-storage numpy==2.1.3 \
     fastapi uvicorn[standard] python-multipart pyloudnorm \
+    google-cloud-storage "numpy==2.1.3" \
     huggingface_hub gradio soxr
 
-# 6) t5x/flaxformer last
+# 7) 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