1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_ner.git synced 2025-06-09 04:10:08 +02:00

add cache

This commit is contained in:
Kai Labusch 2020-08-18 11:49:16 +02:00
parent 7eba58fb11
commit 28fa95b09b
3 changed files with 14 additions and 1 deletions

View file

@ -1,6 +1,8 @@
import os
import logging
from flask import Flask, send_from_directory, redirect, jsonify, request
from flask_caching import Cache
from hashlib import sha256
import html
import json
import torch
@ -16,6 +18,8 @@ app = Flask(__name__)
app.config.from_json('config.json' if not os.environ.get('CONFIG') else os.environ.get('CONFIG'))
cache = Cache(app)
logger = logging.getLogger(__name__)
@ -149,6 +153,10 @@ predictor_store = PredictorStore()
tokenizer = NERTokenizer()
def key_prefix():
return "{}:{}".format(request.path, sha256(str(request.json).encode('utf-8')).hexdigest())
@app.route('/')
def entry():
return redirect("/index.html", code=302)
@ -160,6 +168,7 @@ def get_models():
@app.route('/tokenized', methods=['GET', 'POST'])
@cache.cached(key_prefix=key_prefix)
def tokenized():
raw_text = request.json['text']
@ -173,6 +182,7 @@ def tokenized():
@app.route('/ner-bert-tokens', methods=['GET', 'POST'])
@app.route('/ner-bert-tokens/<model_id>', methods=['GET', 'POST'])
@cache.cached(key_prefix=key_prefix)
def ner_bert_tokens(model_id=None):
raw_text = request.json['text']
@ -198,6 +208,7 @@ def ner_bert_tokens(model_id=None):
@app.route('/ner', methods=['GET', 'POST'])
@app.route('/ner/<model_id>', methods=['GET', 'POST'])
@cache.cached(key_prefix=key_prefix)
def ner(model_id=None):
raw_text = request.json['text']