| from flask import Flask, jsonify, request |
| import pickle |
| import pandas as pd |
|
|
| from utils import predict |
|
|
|
|
| app = Flask(__name__) |
|
|
|
|
| @app.route('/') |
| def index(): |
| return "/model/text – predicts tag for string sample" |
|
|
|
|
| @app.route('/model/text', methods=['POST']) |
| def parse_string(): |
| sample = request.get_data() |
| model = pickle.load(open('models/tags_model.pkl', 'rb')) |
| labelencoder = pickle.load(open('models/tags_encoder.pkl', 'rb')) |
| return predict(sample, model, labelencoder) |
|
|
|
|
| if __name__ == '__main__': |
| app.run(debug=True) |
