INNER CODE UNIT · Python

home

bansalkanav/Ultimate-Data-Science-Toolkit---From-Python-Basics-to-GenerativeAI · Module 5 - MLOPs/2. Model Serving/2. Flask App/app.py:9

def home():
    return render_template("home.html")

@app.route('/prediction', methods=['get', 'post'])
def prediction():
    sl = float(request.form.get("slcm"))
    sw = float(request.form.get("swcm"))
    pl = float(request.form.get("plcm"))
    pw = float(request.form.get("pwcm"))

    data_point = np.array([[sl, sw, pl, pw]])

    model = joblib.load("model/knn.pkl")

    prediction = model.predict(data_point)

    return render_template("output.html", prediction=prediction)

View source record →