MarfinF commited on
Commit
e6a9f3e
·
1 Parent(s): 2cb44f8

- Try again for hf_cache

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -0
Dockerfile CHANGED
@@ -5,6 +5,11 @@ WORKDIR /app
5
  COPY requirements.txt .
6
  RUN pip install --no-cache-dir -r requirements.txt
7
  RUN pip install -U transformers accelerate peft huggingface_hub
 
 
 
 
 
8
  RUN python -c "from transformers import pipeline; pipeline('zero-shot-classification', model='MarfinF/marfin_emotion')"
9
 
10
  COPY . .
 
5
  COPY requirements.txt .
6
  RUN pip install --no-cache-dir -r requirements.txt
7
  RUN pip install -U transformers accelerate peft huggingface_hub
8
+ # ✅ Set environment variable untuk cache model & tokenizers
9
+ ENV TRANSFORMERS_CACHE=/app/cache
10
+ ENV HF_HOME=/app/hf_cache
11
+ RUN mkdir -p /app/cache /app/hf_cache
12
+
13
  RUN python -c "from transformers import pipeline; pipeline('zero-shot-classification', model='MarfinF/marfin_emotion')"
14
 
15
  COPY . .