sathish2352 commited on
Commit
a1cddb1
·
verified ·
1 Parent(s): a3cbe08

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -15,6 +15,8 @@ COPY . .
15
 
16
  # Expose the port Hugging Face expects (default 7860)
17
  EXPOSE 7860
 
 
18
 
19
  # Start the FastAPI app using uvicorn
20
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
15
 
16
  # Expose the port Hugging Face expects (default 7860)
17
  EXPOSE 7860
18
+ ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
19
+ ENV HF_HOME=/tmp/huggingface
20
 
21
  # Start the FastAPI app using uvicorn
22
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]