Update Dockerfile

Dockerfile

@@ -14,20 +14,22 @@ COPY main.py .
 
 # HuggingFace Spaces runs as non-root user 1000
 RUN useradd -m -u 1000 user
-# Pre-download models into the image cache so cold starts are fast
-# (comment out these two lines if you want a smaller image and can tolerate the first-run download)
+
+# Pre-download model into the image cache so cold starts are fast
+# (comment out if you want a smaller image and can tolerate the first-run download)
 RUN python -c "\
 import builtins; builtins.input = lambda p='': 'y'; \
 import os; os.environ['HF_HUB_VERBOSITY']='error'; \
-from transformers import AutoModel, AutoTokenizer; \
-AutoModel.from_pretrained('perplexity-ai/pplx-embed-context-v1-0.6B', trust_remote_code=True); \
-AutoModel.from_pretrained('perplexity-ai/pplx-embed-v1-0.6B', trust_remote_code=True); \
-AutoTokenizer.from_pretrained('perplexity-ai/pplx-embed-v1-0.6B', trust_remote_code=True); \
-print('Models cached.')"
+from transformers import AutoModel; \
+AutoModel.from_pretrained('jinaai/jina-embeddings-v2-base-code', trust_remote_code=True); \
+print('Model cached.')"
 
 USER user
 ENV HOME=/home/user PATH=/home/user/.local/bin:$PATH
 
+# /data is the HuggingFace Spaces persistent volume
+VOLUME ["/data"]
+
 EXPOSE 7860
 
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]