Update Dockerfile

Dockerfile

@@ -1,17 +1,17 @@
-# Use the PyTorch CUDA image
-FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
-
-# Install system build tools for Unsloth
-RUN apt-get update && apt-get install -y \
-    git \
-    build-essential \
-    && rm -rf /var/lib/apt/lists/*
-
-WORKDIR /app
-COPY . .
-
-# Install the ML stack
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Run the training script directly when the Space boots
-CMD ["python", "grpo_train.py"]
+# Use the PyTorch CUDA image
+FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
+
+# Install system build tools for Unsloth
+RUN apt-get update && apt-get install -y \
+    git \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+WORKDIR /app
+COPY . .
+
+# Install the ML stack
+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"]