Update Dockerfile

Dockerfile

@@ -13,6 +13,11 @@ COPY src/ ./src/
 
 RUN pip3 install -r requirements.txt
 
+# --- PRE-DOWNLOAD THE MODEL HERE ---
+RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
+tokenizer = AutoTokenizer.from_pretrained('ma7583/pvs_oracle'); \
+model = AutoModelForCausalLM.from_pretrained('ma7583/pvs_oracle')"
+
 EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health