run_http_api_serve.py

import sys

from flask import Flask, request, jsonify
from flask_cors import CORS
from tokenizer_tools.tagset.NER.BILUO import BILUOSequenceEncoderDecoder

decoder = BILUOSequenceEncoderDecoder()


app = Flask(__name__)
app.config['JSON_AS_ASCII'] = False
app.config['DEBUG'] = True
CORS(app)

from tensorflow.contrib import predictor

export_dir = '/Users/howl/PyCharmProjects/seq2annotation_ner_on_people_daily/evaluate/results/saved_model/1543901916'

predict_fn = predictor.from_saved_model(export_dir)


@app.route("/parse", methods=['GET'])
def single_tokenizer():
    text_msg = request.args.get('q')

    print(text_msg)

    predictions = predict_fn(
        {
            'words': [[i for i in text_msg]],
            'words_len': [len(text_msg)]
        }
    )
    print(predictions['tags'])

    tags_seq = [i.decode() for i in predictions['tags'][0]]

    offset_list = decoder.decode_to_offset(tags_seq)
    print(offset_list)

    response = {
        'text': text_msg,
        'spans': [{'start': i[0], 'end': i[1], 'type': i[2]} for i in offset_list],
        'ents': list({i[2].lower() for i in offset_list})
    }

    return jsonify(response)


if __name__ == "__main__":
    app.run(host='0.0.0.0', port=5000)