Spaces:

idkash1
/

detect-edits-in-ai-generated-text

Sleeping

idkash1 commited on Mar 18

Commit

f510c5d

verified ·

1 Parent(s): 32bfdb7

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -21,7 +21,7 @@ ENV FLASK_APP=app.py
 # Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
-RUN "from transformers import AutoModelForCausalLM, AutoTokenizer; AutoModelForCausalLM.from_pretrained('gpt2-xl', cache_dir='/cache/huggingface'); AutoTokenizer.from_pretrained('gpt2-xl', cache_dir='/cache/huggingface')"
 # Copy the rest of the app
 COPY . .

 # Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
+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')"
 # Copy the rest of the app
 COPY . .