Sudheer-N commited on
Commit
ab642aa
·
1 Parent(s): 72ca74f

Fix model cache permissions

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -0
Dockerfile CHANGED
@@ -2,6 +2,12 @@ FROM python:3.10-slim
2
  WORKDIR /app
3
  COPY requirements.txt .
4
  RUN pip install --no-cache-dir -r requirements.txt
 
5
  COPY app.py .
 
 
 
 
 
6
  EXPOSE 7860
7
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
2
  WORKDIR /app
3
  COPY requirements.txt .
4
  RUN pip install --no-cache-dir -r requirements.txt
5
+
6
  COPY app.py .
7
+
8
+ # Hugging Face model cache
9
+ RUN mkdir -p /app/cache
10
+ ENV TRANSFORMERS_CACHE=/app/cache
11
+
12
  EXPOSE 7860
13
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]