Update Dockerfile

Dockerfile

@@ -19,9 +19,13 @@ ENV HF_HOME=/tmp
 
 # Pre-download model into /tmp to avoid cold start
 RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
-    model_id='rabiyulfahim/Python_coding_model'; \
-    AutoTokenizer.from_pretrained(model_id, cache_dir='/tmp'); \
-    AutoModelForCausalLM.from_pretrained(model_id, cache_dir='/tmp')"
+    from peft import PeftModel; \
+    base_model_id='microsoft/DialoGPT-small'; \
+    adapter_id='rabiyulfahim/Python_coding_model'; \
+    AutoTokenizer.from_pretrained(base_model_id, cache_dir='/tmp'); \
+    base_model = AutoModelForCausalLM.from_pretrained(base_model_id, cache_dir='/tmp'); \
+    PeftModel.from_pretrained(base_model, adapter_id, cache_dir='/tmp')"
+
 
 
 # Expose the correct port (Hugging Face default is 7860)