Update Dockerfile

Dockerfile

@@ -2,11 +2,11 @@
 FROM python:3.10-slim
 
 # Set environment variables to avoid caching issues with Hugging Face
-# ENV HF_HOME=/tmp/huggingface \
-#     TRANSFORMERS_CACHE=/tmp/huggingface/transformers \
-#     HF_DATASETS_CACHE=/tmp/huggingface/datasets \
-#     HUGGINGFACE_HUB_CACHE=/tmp/huggingface/hub \
-#     XDG_CACHE_HOME=/tmp/huggingface
+ENV HF_HOME=/tmp/huggingface \
+    TRANSFORMERS_CACHE=/tmp/huggingface/transformers \
+    HF_DATASETS_CACHE=/tmp/huggingface/datasets \
+    HUGGINGFACE_HUB_CACHE=/tmp/huggingface/hub \
+    XDG_CACHE_HOME=/tmp/huggingface
 
 WORKDIR /app
 
@@ -20,7 +20,10 @@ WORKDIR /app
 COPY requirements.txt ./
 COPY src/ ./src/
 
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt && \
+    python -c "from transformers import AutoTokenizer, AutoModel; \
+    AutoTokenizer.from_pretrained('prd101-wd/phi1_5-bankingqa-merged'); \
+    AutoModel.from_pretrained('prd101-wd/phi1_5-bankingqa-merged')"
 
 EXPOSE 7860