修正docker镜像构建

services/paddle_services/cost_api.py

@@ -0,0 +1,24 @@
+import json
+
+from flask import Flask, request
+from paddlenlp import Taskflow
+
+from utils import process_request
+from . import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES
+
+app = Flask(__name__)
+COST_LIST_SCHEMA = PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + MEDICAL_EXPENSES
+COST = Taskflow('information_extraction', schema=COST_LIST_SCHEMA, model='uie-x-base',
+                task_path='../../model/cost_list_model', layout_analysis=False, precision='fp16')
+
+
+@app.route('/nlp/cost', methods=['POST'])
+@process_request
+def cost():
+    img_path = request.form.get('img_path')
+    layout = request.form.get('layout')
+    return COST({'doc': img_path, 'layout': json.loads(layout)})
+
+
+if __name__ == '__main__':
+    app.run('0.0.0.0', 5004)