mike23415 commited on
Commit
8514dd9
·
verified ·
1 Parent(s): bb32339

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -5,6 +5,10 @@ import pdfkit
5
  import jinja2
6
  import torch
7
  import tempfile
 
 
 
 
8
 
9
  app = Flask(__name__)
10
  CORS(app)
 
5
  import jinja2
6
  import torch
7
  import tempfile
8
+ import os
9
+ os.environ['TRANSFORMERS_CACHE'] = '/app/.cache'
10
+ os.environ['HF_DATASETS_CACHE'] = '/app/.cache'
11
+ os.environ['XDG_CACHE_HOME'] = '/app/.cache'
12
 
13
  app = Flask(__name__)
14
  CORS(app)