forked from yenhao/EmotionDetection_API
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathAPI_Server.py
More file actions
44 lines (32 loc) · 1.15 KB
/
API_Server.py
File metadata and controls
44 lines (32 loc) · 1.15 KB
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
# -*- coding:utf-8 -*-
from flask import Flask, json, request, Response
from flask_restful import reqparse, abort, Api, Resource
import classifier
import json
from pprint import pprint
import multiprocessing as mp
app = Flask(__name__)
api = Api(app)
# data = json.loads(open('input.json').read(), encoding="UTF-8")
# pool = mp.Pool(processes=mp.cpu_count()-1)
# classifier.classifyUsingMatrix(data)
# classifier.classifyUsingMatrixMulti(data)
# api.add_resource(ChuckClassifier, '/chuck/single')
# api.add_resource(ChuckClassifierMulti, '/chuck/couple')
'''
Handle Multi Message using Multi-process
'''
@app.route('/chuck/couple', methods = ['POST'])
def api_message():
if request.headers['Content-Type'] == 'application/json':
# print(request.json)
js = json.dumps(classifier.classifyUsingMatrixMulti(request.json))
return Response(js, status=200, mimetype='application/json')
else:
return "415 Unsupported Data Type"
@app.route('/chuck/test')
def test():
return json.dumps(classifier.classifyUsingMatrixMulti(data))
if __name__ == '__main__':
# app.run(debug=True)
app.run('0.0.0.0',debug = False, port=5678)