Create app.py

app.py

@@ -0,0 +1,28 @@
+from flask import Flask, request, jsonify
+import joblib
+import pandas as pd
+
+app = Flask(__name__)
+
+model = joblib.load("model.pkl")
+
+@app.route("/predict", methods=["POST"])
+def predict():
+    data = request.get_json()
+    df = pd.DataFrame([data])
+    pred = model.predict(df)[0]
+    return jsonify({"prediction": int(pred)})
+
+@app.route("/batch", methods=["POST"])
+def batch():
+    data = request.get_json()
+    df = pd.DataFrame(data["instances"])
+    preds = model.predict(df).tolist()
+    return jsonify({"predictions": preds})
+
+@app.route("/")
+def home():
+    return "Flask API is running."
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)