safihaider commited on
Commit
af43ce7
1 Parent(s): b632c84

permission work

Browse files
Files changed (2) hide show
  1. ChatController.py +3 -0
  2. 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="/path/.cache/huggingface" && export HF_DATASETS_CACHE="/path/.cache/huggingface/datasets" && export TRANSFORMERS_CACHE="/path/.cache/huggingface/models" && gunicorn -b 0.0.0.0:7860 ChatController:app --timeout 700
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