diff --git a/api/app.py b/api/app.py index c3bbad2..4f1a33a 100644 --- a/api/app.py +++ b/api/app.py @@ -31,15 +31,13 @@ def predict(): variable = request.args.get('variable') if variable is None: - print('Variable is None') return jsonify({'errorCode': 404, 'message': 'Variable not found'}) else: result = pipe(variable) result_sorted = sorted(result[0], key=lambda x: x['score'], reverse=True) mapped_result = [] - print("res",result_sorted) - print("mapp", mapped_tags) + for res in result_sorted: mapped_result.append( diff --git a/api/requirements.txt b/api/requirements.txt index 8f934dc..f4a5557 100644 --- a/api/requirements.txt +++ b/api/requirements.txt @@ -3,3 +3,4 @@ Werkzeug==2.2.2 simpletransformers==0.63.9 torch==1.12.1 transformers==4.24.0 +numpy<2 \ No newline at end of file