diff options
Diffstat (limited to 'nlp/scripts/api_squad_online.py')
-rw-r--r-- | nlp/scripts/api_squad_online.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/nlp/scripts/api_squad_online.py b/nlp/scripts/api_squad_online.py index abe3d5f..eba2cd2 100644 --- a/nlp/scripts/api_squad_online.py +++ b/nlp/scripts/api_squad_online.py @@ -13,7 +13,7 @@ import sys from flask import Flask, abort, request, jsonify import os -from global_setting import questions, tokenizer_ch, CUDA_VISIBLE_DEVICES +from global_setting import tokenizer_ch, CUDA_VISIBLE_DEVICES from create_squad_features import get_squad_feature_result @@ -38,6 +38,7 @@ class AI2Flask: except: title = 'Not available' text_origin = request.json['text'] + questions = request.json['questions'] if len(text_origin) > 800: text = text_origin[:800] |