Spaces:
Sleeping
Sleeping
| from flask import Flask, jsonify, request, render_template | |
| import joblib | |
| import numpy as np | |
| app = Flask(__name__) | |
| try: | |
| model = joblib.load('random_forest_model.joblib') | |
| status = 'Loaded' | |
| except: | |
| status = "not loaded" | |
| def health_check(): | |
| return render_template('index.html') | |
| def predict(): | |
| data = np.array([[float(request.form.get('input1', 0)), | |
| float(request.form.get('input2', 0)), | |
| float(request.form.get('input3', 0)), | |
| float(request.form.get('input4', 0)), | |
| float(request.form.get('input5', 0)), | |
| float(request.form.get('input6', 0)), | |
| float(request.form.get('input7', 0))]]) | |
| prediction = model.predict(data).tolist() | |
| return render_template('output.html', prediction = prediction) | |
| if __name__ == '__main__': | |
| app.run(host = '0.0.0.0', debug=False, port = 7860) | |