import numpy as np
from flask import Flask, request, render_template
import pickle
import gradio as gr

app = Flask(__name__)
model = pickle.load(open('weight_pred_model.pkl', 'rb'))

@app.route('/')
def home():
    return render_template('index.html')

@app.route('/getprediction',methods=['POST'])
def getprediction():    

    input = [float(x) for x in request.form.values()]
    final_input = [np.array(input)]
    prediction = model.predict(final_input)

    return render_template('index.html', output='Predicted Weight in KGs :{}'.format(prediction))
   

if __name__ == "__main__":
    app.run(debug=True)