dhruv107 commited on
Commit
632623f
·
1 Parent(s): b3fc83b
Files changed (1) hide show
  1. pretrained.py +3 -0
pretrained.py CHANGED
@@ -8,6 +8,9 @@ from flask import Flask, request, jsonify
8
  from werkzeug.utils import secure_filename
9
  # import logging
10
 
 
 
 
11
 
12
  # # Set up root logger, and add a file handler to root logger
13
  # logging.basicConfig(filename = 'log_file.log',
 
8
  from werkzeug.utils import secure_filename
9
  # import logging
10
 
11
+ import os
12
+ os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
13
+
14
 
15
  # # Set up root logger, and add a file handler to root logger
16
  # logging.basicConfig(filename = 'log_file.log',