mrfirdauss commited on
Commit
39150df
·
verified ·
1 Parent(s): 499c49a

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -2
  2. app.py +2 -2
Dockerfile CHANGED
@@ -6,7 +6,7 @@ RUN pip install --no-cache-dir --upgrade -r /requirements.txt
6
  # Expose the secret SECRET_EXAMPLE at buildtime and use its value as a Bearer token for a curl request
7
  ENV API_KEY=${API_KEY}
8
 
9
- ENV TRANSFORMERS_CACHE=/path/to/writable/directory
10
- RUN mkdir -p /path/to/writable/directory && chmod -R 777 /path/to/writable/directory}
11
 
12
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
6
  # Expose the secret SECRET_EXAMPLE at buildtime and use its value as a Bearer token for a curl request
7
  ENV API_KEY=${API_KEY}
8
 
9
+ ENV TRANSFORMERS_CACHE=/transformers_cache
10
+ RUN mkdir -p /transformers_cache && chmod -R 777 /transformers_cache
11
 
12
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
app.py CHANGED
@@ -6,8 +6,8 @@ import faiss
6
  import os
7
  import requests
8
 
9
- os.environ['TRANSFORMERS_CACHE'] = '/path/to/writable/directory'
10
- os.environ['HF_HOME'] = '/path/to/writable/directory'
11
 
12
  class Generate(BaseModel):
13
  text:str
 
6
  import os
7
  import requests
8
 
9
+ os.environ['TRANSFORMERS_CACHE'] = '/transformers_cache'
10
+ os.environ['HF_HOME'] = '/transformers_cache'
11
 
12
  class Generate(BaseModel):
13
  text:str