Upload Dockerfile with huggingface_hub
Browse files- Dockerfile +8 -0
Dockerfile
CHANGED
|
@@ -2,6 +2,14 @@ FROM python:3.11-slim
|
|
| 2 |
|
| 3 |
WORKDIR /app
|
| 4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
COPY requirements.txt .
|
| 6 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 7 |
|
|
|
|
| 2 |
|
| 3 |
WORKDIR /app
|
| 4 |
|
| 5 |
+
# Create cache directory with proper permissions
|
| 6 |
+
RUN mkdir -p /tmp/hf_cache && chmod 777 /tmp/hf_cache
|
| 7 |
+
|
| 8 |
+
# Set environment variables for Hugging Face cache
|
| 9 |
+
ENV HF_HOME=/tmp/hf_cache
|
| 10 |
+
ENV TRANSFORMERS_CACHE=/tmp/hf_cache
|
| 11 |
+
ENV HF_DATASETS_CACHE=/tmp/hf_cache
|
| 12 |
+
|
| 13 |
COPY requirements.txt .
|
| 14 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 15 |
|