aniudupa commited on
Commit
df3738c
·
verified ·
1 Parent(s): b70960d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +8 -1
Dockerfile CHANGED
@@ -6,6 +6,10 @@ FROM python:3.9-slim
6
  RUN useradd -m -u 1000 user
7
  USER user
8
  ENV PATH="/home/user/.local/bin:$PATH"
 
 
 
 
9
 
10
  WORKDIR /app
11
 
@@ -18,6 +22,9 @@ RUN pip install -U langchain-huggingface
18
  # Step 6: Install faiss-gpu separately (if needed)
19
  RUN pip install --no-cache-dir faiss-gpu
20
 
 
 
 
21
 
22
  COPY --chown=user . /app
23
- CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
6
  RUN useradd -m -u 1000 user
7
  USER user
8
  ENV PATH="/home/user/.local/bin:$PATH"
9
+ ENV HF_HOME=/home/user/huggingface
10
+ ENV TRANSFORMERS_CACHE=/home/user/huggingface/transformers
11
+ ENV SENTENCE_TRANSFORMERS_HOME=/home/user/sentence_transformers
12
+ ENV TORCH_HOME=/home/user/torch
13
 
14
  WORKDIR /app
15
 
 
22
  # Step 6: Install faiss-gpu separately (if needed)
23
  RUN pip install --no-cache-dir faiss-gpu
24
 
25
+ RUN chmod -R 777 /home/user
26
+
27
+ EXPOSE 7860
28
 
29
  COPY --chown=user . /app
30
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--reload"]