File size: 615 Bytes
999ae17
 
 
 
 
 
 
 
 
 
 
 
5433f7c
 
 
 
999ae17
 
5433f7c
999ae17
7d12025
5433f7c
999ae17
 
 
 
 
5433f7c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from flask import Flask, request, jsonify
import pickle
import numpy as np
import logging

app = Flask(__name__)

log = logging.getLogger('werkzeug')
log.setLevel(logging.ERROR)

model = pickle.load(open("lead_conversion_model.pkl","rb"))

@app.route("/")
def home():
    return "Lead Conversion API is running!"

@app.route("/predict", methods=["POST"])
def predict():
    
    data = request.json["data"]
    prediction = model.predict(np.zeros((1, model.n_features_in_)))
    
    return jsonify({
        "prediction": int(prediction[0])
    })

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)