|
from flask import Flask, request, redirect, url_for, flash, jsonify |
|
import numpy as np |
|
import pickle as p |
|
import json |
|
|
|
|
|
app = Flask(__name__) |
|
|
|
@app.route("/") |
|
def home(): |
|
return "<h1>Lyrics Classifier</h1>" |
|
|
|
@app.route('/api/', methods=['POST']) |
|
def makecalc(): |
|
data = request.get_json(force=True) |
|
prediction = np.array2string(model.predict(data['lyric'])) |
|
print(prediction) |
|
return jsonify(prediction) |
|
|
|
|
|
if __name__ == '__main__': |
|
modelfile = 'model.pickle' |
|
model = p.load(open(modelfile, 'rb')) |
|
app.run(debug=True, host='0.0.0.0') |
|
|
