Upload Dockerfile

Dockerfile

@@ -1,22 +1,31 @@
-FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
-
-ENV DEBIAN_FRONTEND=noninteractive
-RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
+FROM huggingface/transformers-pytorch-gpu:4.46.3
 
 WORKDIR /app
 
-# Install Unsloth and training dependencies
-RUN pip install --no-cache-dir \
-    "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" \
-    && pip install --no-cache-dir --no-deps xformers \
-    && pip install --no-cache-dir \
-    trl peft accelerate bitsandbytes huggingface_hub safetensors
+# System dependencies
+RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
+
+# Add HuggingFace unprivileged user bounds
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    PYTHONUNBUFFERED=1 \
+    FORCE_CUDA="1"
+
+WORKDIR $HOME/app
+
+# Dependencies
+COPY --chown=user:user requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy training script and dataset
-COPY train.py .
-COPY aegis_training_data_500.json .
+# Secure Application Code
+COPY --chown=user:user aegis_env aegis_env/
+COPY --chown=user:user hf_training hf_training/
+COPY --chown=user:user scripts scripts/
+COPY --chown=user:user aegis_training_data_500.json .
 
 EXPOSE 7860
 
-# -u for unbuffered stdout so logs appear in real time in HF Space console
-CMD ["python", "-u", "train.py"]
+# Run training entry point
+CMD ["python3", "hf_training/train.py"]