Bijoy09 commited on
Commit
4e9fdd0
1 Parent(s): 7bd1f8b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -13,7 +13,10 @@ RUN pip install --no-cache-dir -r requirements.txt
13
  EXPOSE 7860
14
 
15
  # Set environment variable for Hugging Face cache directory
16
- ENV HF_HOME="/app/.cache"
 
 
 
17
 
18
  # Command to run the FastAPI app with Uvicorn
19
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
13
  EXPOSE 7860
14
 
15
  # Set environment variable for Hugging Face cache directory
16
+ ENV TRANSFORMERS_CACHE="/app/cache"
17
+
18
+ # Create the cache directory
19
+ RUN mkdir -p /app/cache
20
 
21
  # Command to run the FastAPI app with Uvicorn
22
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]