Spaces:
Paused
Paused
safihaider
commited on
Commit
•
af43ce7
1
Parent(s):
b632c84
permission work
Browse files- ChatController.py +3 -0
- Dockerfile +1 -1
ChatController.py
CHANGED
@@ -1,6 +1,9 @@
|
|
1 |
from flask import Flask, request
|
2 |
from flask_cors import CORS, cross_origin
|
3 |
from ChatService import ChatService
|
|
|
|
|
|
|
4 |
|
5 |
app = Flask(__name__)
|
6 |
CORS(app)
|
|
|
1 |
from flask import Flask, request
|
2 |
from flask_cors import CORS, cross_origin
|
3 |
from ChatService import ChatService
|
4 |
+
import os
|
5 |
+
|
6 |
+
os.environ['TRANSFORMERS_CACHE'] = './cache/'
|
7 |
|
8 |
app = Flask(__name__)
|
9 |
CORS(app)
|
Dockerfile
CHANGED
@@ -11,7 +11,7 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
|
11 |
|
12 |
COPY . .
|
13 |
|
14 |
-
CMD export HF_HOME="
|
15 |
|
16 |
# FROM python:3
|
17 |
|
|
|
11 |
|
12 |
COPY . .
|
13 |
|
14 |
+
CMD export HF_HOME="./cache" && export HF_DATASETS_CACHE="./cache" && export TRANSFORMERS_CACHE="./cache" && gunicorn -b 0.0.0.0:7860 ChatController:app --timeout 700
|
15 |
|
16 |
# FROM python:3
|
17 |
|