Update Dockerfile

Dockerfile

@@ -1,17 +1,23 @@
-# Use the PyTorch CUDA image
-FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
+FROM python:3.11-slim
 
-# Install system build tools for Unsloth
-RUN apt-get update && apt-get install -y \
-    git \
-    build-essential \
-    && rm -rf /var/lib/apt/lists/*
+RUN apt-get update \
+ && apt-get install -y --no-install-recommends curl \
+ && rm -rf /var/lib/apt/lists/*
+
+RUN useradd -m -u 1000 user
+ENV USER=user
+ENV HOME=/home/user
 
 WORKDIR /app
-COPY . .
 
-# Install the ML stack
+COPY pyproject.toml requirements.txt ./
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Run the training script directly when the Space boots
-CMD ["/bin/bash", "-c", "python -m http.server 7860 & python grpo_train.py"]
+COPY . .
+RUN pip install --no-cache-dir --no-deps .
+
+RUN chmod +x apps/start.sh
+
+EXPOSE 8000
+
+CMD ["./apps/start.sh"]