idkash1 commited on
Commit
95088a5
·
verified ·
1 Parent(s): f510c5d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +0 -2
Dockerfile CHANGED
@@ -21,8 +21,6 @@ ENV FLASK_APP=app.py
21
  # Install dependencies
22
  RUN pip install --no-cache-dir -r requirements.txt
23
 
24
- RUN python -c "from transformers import AutoModelForCausalLM, AutoTokenizer; AutoModelForCausalLM.from_pretrained('gpt2-xl', cache_dir='/cache/huggingface'); AutoTokenizer.from_pretrained('gpt2-xl', cache_dir='/cache/huggingface')"
25
-
26
  # Copy the rest of the app
27
  COPY . .
28
 
 
21
  # Install dependencies
22
  RUN pip install --no-cache-dir -r requirements.txt
23
 
 
 
24
  # Copy the rest of the app
25
  COPY . .
26