Update Dockerfile

Dockerfile

@@ -15,14 +15,18 @@ RUN curl -fsSL https://deb.nodesource.com/setup_20.x | bash - \
     && npm install -g yarn pnpm typescript ts-node nodemon \
     && apt-get clean && rm -rf /var/lib/apt/lists/*
 
-# ---- Python deps ----
+# ---- Python deps (from PyPI) ----
 RUN pip install --no-cache-dir \
     requests ipython huggingface_hub transformers \
-    torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu \
     numpy pandas scipy scikit-learn matplotlib \
     flask fastapi uvicorn jupyter \
     black ruff pylint
 
+# ---- PyTorch CPU-only (from PyTorch index) ----
+RUN pip install --no-cache-dir \
+    torch torchvision torchaudio \
+    --extra-index-url https://download.pytorch.org/whl/cpu
+
 # ---- create user WITH PASSWORD + sudo ----
 RUN useradd -m -u 1000 -s /bin/bash user \
     && echo "user:user123" | chpasswd \
@@ -39,7 +43,6 @@ WORKDIR /home/user
 # ---- workspace + code-server extensions ----
 RUN mkdir -p /home/user/workspace /home/user/.local/share/code-server
 
-# Install useful extensions (Codeium AI, Python, etc.)
 RUN code-server --install-extension ms-python.python \
     && code-server --install-extension esbenp.prettier-vscode \
     && code-server --install-extension bradlc.vscode-tailwindcss \