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