Upload Dockerfile with huggingface_hub

Dockerfile

@@ -0,0 +1,38 @@
+FROM python:3.11-slim
+
+# Install build tools for torch.compile
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create a non-root user with home directory for cache
+RUN useradd -m -u 1000 user
+USER user
+
+WORKDIR /home/user/app
+
+# Set cache directory to user's home
+ENV HOME=/home/user
+ENV HF_HOME=/home/user/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/home/user/.cache/huggingface/transformers
+ENV HF_DATASETS_CACHE=/home/user/.cache/huggingface/datasets
+
+# Install dependencies as user (use --user flag)
+RUN pip install --user --no-cache-dir \
+    torch \
+    transformers>=4.48.0 \
+    datasets \
+    evaluate \
+    seqeval \
+    accelerate>=1.0.0 \
+    huggingface_hub
+
+# Add user's pip bin to PATH
+ENV PATH="/home/user/.local/bin:${PATH}"
+
+# Copy training script and data
+COPY --chown=user:user train.py .
+COPY --chown=user:user data/ ./data/
+
+# Run training
+CMD ["python", "train.py"]