-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathserver.py
More file actions
33 lines (24 loc) · 853 Bytes
/
Copy pathserver.py
File metadata and controls
33 lines (24 loc) · 853 Bytes
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
from flask import Flask, render_template, request, jsonify
import tensorflow as tf
app = Flask(__name__)
model = tf.keras.models.load_model('model')
@app.route('/', methods=['GET'])
def process():
return render_template('index.html')
@app.route('/api/v1/check', methods=['GET'])
def api_check():
if 'message' in request.args:
message = request.args['message']
else:
return jsonify({
'result': -1,
'msg': 'No message provided. Please specify a message to be checked.'
}), 422
# No hate speech = 0 / Hate speech = 1
result = 1 if model.predict([str(message)]) > 0.5 else 0
return jsonify({
'result': result,
'msg': 'Message is ' + ('' if result == 1 else 'not ') + 'a hate speech.'
}), 200
if __name__ == '__main__':
app.run(host='0.0.0.0', port=80)