From d5dde5aaf4f2d2cbccd014b5aa6a97872d345bf9 Mon Sep 17 00:00:00 2001 From: Michelle <mi00354@surrey.ac.uk> Date: Fri, 21 May 2021 23:11:21 +0100 Subject: [PATCH] Added database --- app.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/app.py b/app.py index 725a50c..e551fd3 100644 --- a/app.py +++ b/app.py @@ -2,15 +2,20 @@ import numpy as np from flask import Flask, request, jsonify, render_template import pickle import json +from flask_pymongo import PyMongo app = Flask(__name__) #Initialize the flask App + +mongodb_client = PyMongo(app, uri="mongodb://localhost:27017/pred_records") +db = mongodb_client.db + model = pickle.load(open('model.pickle', 'rb')) mlb = pickle.load(open('mlb.pickle','rb')) vectorizer = pickle.load(open('vectorizer.pickle','rb')) @app.route('/') def home(): - return render_template('index.html') + return render_template('index.ejs') @app.route('/predict',methods=['POST']) def predict(): @@ -18,10 +23,12 @@ def predict(): vector = vectorizer.transform([reqInput]) prediction = model.predict(vector) output = json.dumps(mlb.inverse_transform(prediction)[0]) + db.preds.insert_one({'plot': reqInput, 'prediction': output}) return { "predictionText":output, "predictionRaw": json.dumps(prediction[0].tolist()) } + if __name__ == "__main__": app.run(threaded=True) \ No newline at end of file -- GitLab