DeepImagix commited on
Commit
b5d454f
·
verified ·
1 Parent(s): 13042b1

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -26,7 +26,10 @@ COPY ./main.py /code/main.py
26
  RUN pip install --upgrade pip
27
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
28
 
29
- # Expose the port used by Uvicorn / Hugging Face Spaces
 
 
 
30
  EXPOSE 7860
31
 
32
  # Start the app
 
26
  RUN pip install --upgrade pip
27
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
28
 
29
+ # Pre-download the sentence-transformers model
30
+ RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')"
31
+
32
+ # Expose the port used by Uvicorn
33
  EXPOSE 7860
34
 
35
  # Start the app