Ivan Shelonik commited on
Commit
543f7b1
·
1 Parent(s): 0555843

upd: load type

Browse files
Files changed (1) hide show
  1. api_server.py +1 -1
api_server.py CHANGED
@@ -13,7 +13,7 @@ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
13
  from tensorflow import keras
14
  from flask import Flask, jsonify, request
15
 
16
- load_type = ''
17
  """
18
  local;
19
  remote_hub_download; - /cache error even using TRANSFORMERS_CACHE & cache_dir to local folder
 
13
  from tensorflow import keras
14
  from flask import Flask, jsonify, request
15
 
16
+ load_type = 'remote_hub_from_pretrained'
17
  """
18
  local;
19
  remote_hub_download; - /cache error even using TRANSFORMERS_CACHE & cache_dir to local folder