mindchain commited on
Commit
bf53f58
Β·
verified Β·
1 Parent(s): 306c5f0

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +21 -0
Dockerfile ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.11-slim
2
+
3
+ WORKDIR /app
4
+
5
+ # Install system dependencies
6
+ RUN apt-get update && apt-get install -y \
7
+ git \
8
+ && rm -rf /var/lib/apt/lists/*
9
+
10
+ # Install Python dependencies
11
+ COPY requirements.txt .
12
+ RUN pip install --no-cache-dir -r requirements.txt
13
+
14
+ # Copy training script
15
+ COPY train_arithmetic.py .
16
+
17
+ # Set HF token from secret
18
+ ENV HF_TOKEN=${HF_TOKEN}
19
+
20
+ # Run training
21
+ CMD ["python", "train_arithmetic.py"]