Update Dockerfile

Dockerfile

@@ -1,41 +1,29 @@
-# ---- Base
 FROM huggingface/autotrain-advanced:latest
 
-# ---- Env (faster/cleaner installs + nicer HF downloads)
+# Speed & smaller layers
 ENV PIP_NO_CACHE_DIR=1 \
     PIP_DISABLE_PIP_VERSION_CHECK=1 \
     HF_HUB_ENABLE_HF_TRANSFER=1 \
     HF_HUB_ENABLE_QUICK_EXTRACT=1
 
-# ---- System deps (optional but handy)
-RUN apt-get update && apt-get install -y --no-install-recommends git && rm -rf /var/lib/apt/lists/*
-
-# ---- Python deps
-# NOTE: Install Transformers from source to ensure gpt_oss is recognized.
-#       Keep tokenizers/accelerate fresh as well.
+# Python deps (install Transformers from ZIP so we don't need git)
 RUN python -m pip install --upgrade pip setuptools wheel && \
-    pip install --no-cache-dir -U \
-        autotrain-advanced \
-        "git+https://github.com/huggingface/transformers.git" \
-        "tokenizers>=0.20.0" \
-        "accelerate>=1.0.0" \
-        kernels \
-        huggingface_hub
+    pip install -U autotrain-advanced && \
+    pip install -U "transformers @ https://github.com/huggingface/transformers/archive/refs/heads/main.zip" \
+                  "tokenizers>=0.20.0" "accelerate>=1.0.0" kernels huggingface_hub
 
-# ---- (GPU) Choose the right Torch wheel for your host. Uncomment ONE:
-# CUDA 12.1 example:
-# RUN pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu121 torch torchvision torchaudio
-# CUDA 11.8 example:
-# RUN pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu118 torch torchvision torchaudio
+# ---- (GPU) Choose ONE Torch line that matches your host ----
+# CUDA 12.1:
+# RUN pip install --index-url https://download.pytorch.org/whl/cu121 torch torchvision torchaudio
+# CUDA 11.8:
+# RUN pip install --index-url https://download.pytorch.org/whl/cu118 torch torchvision torchaudio
 # CPU-only:
-# RUN pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cpu torch torchvision torchaudio
+# RUN pip install --index-url https://download.pytorch.org/whl/cpu torch torchvision torchaudio
 
-# ---- Health check: ensure AutoConfig sees gpt_oss at build time (optional)
+# Optional sanity check at build time
 # RUN python - <<'PY'
-# from transformers import AutoConfig, __version__
-# print("Transformers:", __version__)
-# AutoConfig.from_pretrained("openai/gpt-oss-2b")  # or your target checkpoint
-# print("OK: gpt_oss recognized")
+# from transformers import AutoConfig, __version__; print("Transformers:", __version__)
+# AutoConfig.from_pretrained("openai/gpt-oss-2b"); print("OK: gpt_oss recognized")
 # PY
 
 EXPOSE 7860