|
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) |
