revert

Dockerfile

@@ -1,17 +1,38 @@
 # Base image with CUDA 12.8, cuDNN 9, PyTorch 2.7.0
 FROM pytorch/pytorch:2.7.0-cuda12.8-cudnn9-runtime
 
-# Install system tools (optional, like git, wget)
-RUN apt-get update && apt-get install -y \
-    git \
-    && rm -rf /var/lib/apt/lists/*
-
 # Set working directory
-WORKDIR /app
+WORKDIR /usr/src/app
+
+# Install system dependencies, then clean up
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends git wget && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
 
-# Copy and install Python dependencies
-COPY requirements.txt .
+# Copy application files
+COPY . .
+
+# Install Python dependencies
+RUN pip install --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Default command: open Python shell
-CMD ["python"]
+# Make script executable and move it to PATH
+RUN install -m 755 sft.py /usr/local/bin/sft
+
+# Create a non-root user
+RUN useradd --create-home appuser
+RUN chown -R appuser:appuser /usr/src/app
+
+# Use non-root user
+USER appuser
+
+# Set environment variables
+ENV XDG_CACHE_HOME=/usr/src/app/cache
+
+# Expose the Gradio port
+EXPOSE 7860
+ENV GRADIO_SERVER_NAME="0.0.0.0"
+
+# Start the application
+CMD ["python", "app.py"]