Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM debian:12-slim
+FROM python:3.10-slim
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV VENV_PATH=/BitNet/venv
@@ -11,10 +11,6 @@ RUN apt-get update && apt-get install -y \
     git \
     cmake \
     build-essential \
-    python3 \
-    python3-pip \
-    python3-venv \
-    python-is-python3 \
     ca-certificates \
     && rm -rf /var/lib/apt/lists/*
 
@@ -26,25 +22,25 @@ COPY . /BitNet
 WORKDIR /BitNet
 
 # Create virtual environment
-RUN python3 -m venv $VENV_PATH
+RUN python -m venv $VENV_PATH
 
 # Upgrade pip and install dependencies inside venv
-RUN python3 -m pip install --upgrade pip
-RUN python3 -m pip install -r requirements.txt
-RUN python3 -m pip install huggingface_hub
+RUN python -m pip install --upgrade pip
+RUN python -m pip install -r requirements.txt
+RUN python -m pip install huggingface_hub
 
 # Optional: Hugging Face token
 ARG HF_TOKEN
 RUN huggingface-cli login --token $HF_TOKEN || true
 
 # Download model
-RUN python3 -m huggingface-cli download HF1BitLLM/Llama3-8B-1.58-100B-tokens --local-dir models/Llama3-8B-1.58-100B-tokens
+RUN python -m huggingface-cli download HF1BitLLM/Llama3-8B-1.58-100B-tokens --local-dir models/Llama3-8B-1.58-100B-tokens
 
 # Setup environment / quantization
-RUN python3 -m setup_env -md models/Llama3-8B-1.58-100B-tokens -q i2_s
+RUN python -m setup_env -md models/Llama3-8B-1.58-100B-tokens -q i2_s
 
 # Build C++ runtime
 RUN mkdir -p build && cd build && cmake .. && make -j$(nproc)
 
 # Default command
-CMD ["python3", "app.py"]
+CMD ["python", "app.py"]