Spaces:
Sleeping
Sleeping
| 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")) | |
| def home(): | |
| return "Lead Conversion API is running!" | |
| 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) |