app.py

from flask import Flask, render_template, request, redirect
import pickle

app = Flask(__name__)
Filename = 'model/model_tree.pkl'
with open(Filename, 'rb') as file:  
    model = pickle.load(file)

@app.route('/')
def index_page():
    print(model)
    return render_template('index.html')

@app.route('/predict', methods=['POST', 'GET'])
def predict_logic():
    
    if request.method == 'POST':
        acousticness = float(request.form.get('acousticness'))
        danceability = float(request.form.get('danceability'))
        energy = float(request.form.get('energy'))
        instrumentalness = float(request.form.get('instrumentalness'))
        liveness = float(request.form.get('liveness'))
        speechiness = float(request.form.get('speechiness'))                  
        tempo = float(request.form.get('tempo'))
        valence = float(request.form.get('valence'))
    #print(acousticness,danceability,energy,instrumentalness,liveness,speechiness,tempo,valence)
    #print([[acousticness,danceability,energy,instrumentalness,liveness,speechiness,tempo,valence]])
    pred_name = model.predict([[acousticness,danceability,energy,instrumentalness,liveness,speechiness,tempo,valence]]).tolist()[0]
    if  pred_name == 'Rock':
        pred = 'rock.gif'
    else:
        pred = 'hip-hop.gif'
    return render_template('index.html', prediction=pred, pred_name=pred_name)

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