Update Dockerfile

Dockerfile

@@ -16,17 +16,11 @@ COPY main.py .
 RUN useradd -m -u 1000 user
 
 # Pre-download model into the image cache so cold starts are fast.
-# The || true ensures a network failure during build doesn't break the image —
-# the model will simply be downloaded on first startup instead.
+# The || true ensures a network failure during build doesn't break the image.
 RUN python -c "\
-import builtins; builtins.input = lambda p='': 'y'; \
 import os; os.environ['HF_HUB_VERBOSITY'] = 'error'; \
-from transformers import AutoModel; \
-model = AutoModel.from_pretrained( \
-    'jinaai/jina-embeddings-v2-base-code', \
-    trust_remote_code=True, \
-    low_cpu_mem_usage=True, \
-); \
+from sentence_transformers import SentenceTransformer; \
+SentenceTransformer('jinaai/jina-embeddings-v2-base-code', trust_remote_code=True); \
 print('Model cached.')" || echo "Model pre-cache skipped — will download at runtime."
 
 USER user