Update Dockerfile

Dockerfile

@@ -13,6 +13,13 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Set a writable cache directory for Hugging Face models
+ENV HF_HOME=/app/hf_cache
+RUN mkdir -p /app/hf_cache
+
+# Pre-download T5-Base model and tokenizer
+RUN python -c "from transformers import T5Tokenizer, T5ForConditionalGeneration; T5Tokenizer.from_pretrained('t5-base'); T5ForConditionalGeneration.from_pretrained('t5-base')"
+
 # Copy the entire application
 COPY . .