Update Dockerfile

Dockerfile

@@ -12,7 +12,7 @@ RUN mkdir -p /cache/huggingface /cache/matplotlib \
     && chmod -R 777 /cache/huggingface /cache/matplotlib
 
 # Set environment variables for Hugging Face and Matplotlib 
-ENV HF_HOME=/cache/huggingfacec
+ENV HF_HOME=/cache/huggingface
 ENV MPLCONFIGDIR=/cache/matplotlib
 
 # Flask
@@ -21,6 +21,8 @@ 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 . .