kaiser/run_rest_service.py at master · machinereading/kaiser · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

import os
import sys
sys.path.append('../')

from flask import Flask
from flask_restful import Api, Resource, reqparse

from kaiser import parser

app = Flask(__name__)
api = Api(app)

import jpype
jpype.attachThreadToJVM()

# In[1]:


class KaiserSerive(Resource):
    def __init__(self):
#         model_dir = '/disk_4/resource/models'
#         model_path = '/home/hahmyg/FrameNet-RE/models/bert_ko_framenet_model.pt'
#         model_path = '/disk/data/models/bert_ko_framenet_model.pt'
        model_path = '/home/hahmyg/bert_ko_framenet_model.pt'
        self.parser = parser.ShallowSemanticParser(model_path=model_path)
    def post(self):
        try:
            req_parser = reqparse.RequestParser()
            req_parser.add_argument('text', type=str)
            req_parser.add_argument('sent_id', type=str)
            req_parser.add_argument('result_format', type=str)
            args = req_parser.parse_args()
            print(args)
            if not args['sent_id']:
                sent_id = False
            else:
                sent_id = args['sent_id']
            if not args['result_format']:
                result_format = 'graph'
            else:
                result_format = args['result_format']

            result = self.parser.parser(args['text'], sent_id=sent_id, result_format=result_format)

            return result, 200
        except KeyboardInterrupt:
            raise
        except Exception as e:
            return {'error':str(e)}

api.add_resource(KaiserSerive, '/frame_parser')
# api.add_resource(FrameNetRE, '/FRDF/')
app.run(debug=True, host='0.0.0.0', port=1224)