Natthathida commited on
Commit
a7ba413
·
verified ·
1 Parent(s): 09973cb

update TRANSFORMERS_CACHE

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from PIL import Image
8
  from flask_cors import CORS
9
  import os
10
 
11
- os.environ['HF_HOME'] = '/generate_caption/cache/'
12
 
13
  app = Flask(__name__)
14
  CORS(app)
 
8
  from flask_cors import CORS
9
  import os
10
 
11
+ os.environ['TRANSFORMERS_CACHE'] = '/generate_caption/cache/'
12
 
13
  app = Flask(__name__)
14
  CORS(app)