Deploy Flask backend to ExtraaLeanBackend

Dockerfile

@@ -0,0 +1,7 @@
+FROM python:3.9-slim
+WORKDIR /app
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+COPY app.py best_model.pkl .
+EXPOSE 7860
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,25 @@
+from flask import Flask, request, jsonify
+import joblib, pandas as pd, os
+
+app = Flask(__name__)
+# Load the serialized preprocessing+model pipeline
+model = joblib.load("best_model.pkl")
+
+@app.route("/predict", methods=["POST"])
+def predict():
+    # Expect JSON body with feature keys matching training data
+    data = request.get_json()
+    df = pd.DataFrame([data])
+    # Predict class and probability
+    pred = int(model.predict(df)[0])
+    prob = float(model.predict_proba(df)[0, 1])
+    return jsonify({"prediction": pred, "probability": prob})
+
+@app.route("/health", methods=["GET"])
+def health():
+    return jsonify({"status": "ok"})
+
+if __name__ == "__main__":
+    # Use PORT env var if provided, otherwise default to 7860
+    port = int(os.environ.get("PORT", 7860))
+    app.run(host="0.0.0.0", port=port)

best_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:32695d2da09f00f6bb8bdc5e845aee2a82991205c7749a18144d249710604c95
+size 5700267

huggingface.yml

@@ -0,0 +1,2 @@
+app_port: 7860
+sdk: docker

requirements.txt

@@ -0,0 +1,5 @@
+flask
+joblib
+pandas
+scikit-learn
+numpy