Spaces:
Sleeping
Sleeping
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) |