Update Dockerfile

Dockerfile

@@ -1,10 +1,24 @@
 FROM python:3.8-slim
 
-RUN pip install --upgrade pip setuptools wheel
-RUN pip install torch transformers datasets deepspeed vllm accelerate unsloth
-RUN pip install huggingface_hub
+# Set environment variables
+ENV PYTHONUNBUFFERED=1 \
+    DEBIAN_FRONTEND=noninteractive
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
+    pip install --no-cache-dir -r requirements.txt
 
 WORKDIR /app
 COPY . /app
 
+# Pre-create directories needed by training script
+RUN mkdir -p /app/checkpoints /app/logs
+
 CMD ["python", "train.py"]