File size: 455 Bytes
ee25e02
c19f36d
 
ead4532
0c0bd4a
 
ead4532
0c0bd4a
 
ead4532
c19f36d
 
18b0582
 
c19f36d
ead4532
0c0bd4a
c19f36d
18b0582
b3078f3
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# Use Python 3.10-slim as the base image
FROM python:3.10-slim

# Set Hugging Face cache dir to a writable path
ENV HF_HOME=/tmp/hf_cache

# Create the cache directory and give full permissions
RUN mkdir -p /tmp/hf_cache && chmod -R 777 /tmp/hf_cache

# Set working directory
WORKDIR /code

# Copy files to the container
COPY . .

# Install dependencies
RUN pip install --no-cache-dir -r requirements.txt

# Run training script
CMD ["python", "train.py"]