Ivan Shelonik commited on
Commit
c6fdf85
·
1 Parent(s): 4acae3e

upd: update load type

Browse files
Files changed (1) hide show
  1. api_server.py +1 -1
api_server.py CHANGED
@@ -8,7 +8,7 @@ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
8
  from tensorflow import keras
9
  from flask import Flask, jsonify, request
10
 
11
- load_type = ''
12
  """
13
  local,
14
  remote_hub_download - /cache error even using TRANSFORMERS_CACHE to root folder
 
8
  from tensorflow import keras
9
  from flask import Flask, jsonify, request
10
 
11
+ load_type = 'remote_hub_from_pretrained'
12
  """
13
  local,
14
  remote_hub_download - /cache error even using TRANSFORMERS_CACHE to root folder