Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files- Dockerfile +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=/
|
| 10 |
-
RUN mkdir -p
|
| 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'] = '/
|
| 10 |
-
os.environ['HF_HOME'] = '/
|
| 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
|