fix deprecation warning
Browse files- Dockerfile +1 -1
- app.py +0 -1
Dockerfile
CHANGED
|
@@ -7,7 +7,7 @@ RUN apt-get install git-lfs
|
|
| 7 |
RUN git lfs install
|
| 8 |
RUN pip install --no-cache-dir --upgrade -r /requirements.txt
|
| 9 |
|
| 10 |
-
ENV TRANSFORMERS_CACHE=/transformers_cache
|
| 11 |
RUN mkdir -p /transformers_cache && chmod -R 777 /transformers_cache
|
|
|
|
| 12 |
|
| 13 |
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
|
|
|
|
| 7 |
RUN git lfs install
|
| 8 |
RUN pip install --no-cache-dir --upgrade -r /requirements.txt
|
| 9 |
|
|
|
|
| 10 |
RUN mkdir -p /transformers_cache && chmod -R 777 /transformers_cache
|
| 11 |
+
ENV HF_HOME=/transformers_cache
|
| 12 |
|
| 13 |
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
|
app.py
CHANGED
|
@@ -8,7 +8,6 @@ from PyPDF2 import PdfReader
|
|
| 8 |
import requests
|
| 9 |
import classificator
|
| 10 |
|
| 11 |
-
os.environ['TRANSFORMERS_CACHE'] = '/transformers_cache'
|
| 12 |
os.environ['HF_HOME'] = '/transformers_cache'
|
| 13 |
|
| 14 |
|
|
|
|
| 8 |
import requests
|
| 9 |
import classificator
|
| 10 |
|
|
|
|
| 11 |
os.environ['HF_HOME'] = '/transformers_cache'
|
| 12 |
|
| 13 |
|