ananyagm
commited on
Commit
·
5061140
1
Parent(s):
6946df3
Context-Informed-Descriptions
Browse files- Dockerfile +7 -2
- main.py +0 -1
Dockerfile
CHANGED
|
@@ -6,8 +6,13 @@ COPY ./requirements.txt /code/requirements.txt
|
|
| 6 |
|
| 7 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 8 |
|
| 9 |
-
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
-
|
|
|
|
|
|
|
| 12 |
|
| 13 |
CMD ["gunicorn", "-b", "0.0.0.0:7860", "main:app"]
|
|
|
|
| 6 |
|
| 7 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 8 |
|
| 9 |
+
RUN useradd -m -u 1000 user
|
| 10 |
+
USER user
|
| 11 |
+
ENV HOME=/home/user \
|
| 12 |
+
PATH=/home/user/.local/bin:$PATH
|
| 13 |
|
| 14 |
+
WORKDIR $HOME/app
|
| 15 |
+
|
| 16 |
+
COPY --chown=user . $HOME/app
|
| 17 |
|
| 18 |
CMD ["gunicorn", "-b", "0.0.0.0:7860", "main:app"]
|
main.py
CHANGED
|
@@ -1,6 +1,5 @@
|
|
| 1 |
import json
|
| 2 |
import os
|
| 3 |
-
os.environ['TRANSFORMERS_CACHE'] = './.cache'
|
| 4 |
import tiktoken
|
| 5 |
import requests
|
| 6 |
from flask_cors import CORS, cross_origin
|
|
|
|
| 1 |
import json
|
| 2 |
import os
|
|
|
|
| 3 |
import tiktoken
|
| 4 |
import requests
|
| 5 |
from flask_cors import CORS, cross_origin
|