killswitch009's picture
Update app.py
7d12025 verified
Raw
History Blame Contribute Delete
615 Bytes
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)