def start_server(): from bert_base.server import BertServer from bert_base.server.helper import get_run_args args = get_run_args() server = BertServer(args) server.start() server.join()
def start_server(): from bert_base.server import BertServer from bert_base.server.helper import get_run_args args = get_run_args() print(args) import ipdb #ipdb.set_trace() server = BertServer(args) server.start() server.join()
app = Flask(__name__) from bert_base.server.helper import get_run_args import os import helpers class BertMsg(): def __init__(self, msg): self.msg = msg #------------------------------------------------------------------------------------------ ##获取模型运行参数 args = get_run_args() num_labels, label2id, id2label = helpers.init_predict_var( args.classify_model_pb_dir) classify_graph_path = os.path.join(args.classify_model_pb_dir, "classification_model.pb") with tf.gfile.GFile(classify_graph_path, 'rb') as f: classify_graph_def = tf.GraphDef() classify_graph_def.ParseFromString(f.read()) classify_bertWorker = BertWorker(args, args.device_map, classify_graph_path, "CLASS", id2label, "", "") msg = [] classify_bert = BertMsg(msg) classify_estimator = classify_bertWorker.get_estimator(tf, classify_graph_def, "CLASS")