Upload 4 files

Dockerfile

@@ -1,21 +1,6 @@
 FROM python:3.9-slim
-
 WORKDIR /app
-
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    software-properties-common \
-    git \
-    && rm -rf /var/lib/apt/lists/*
-
-COPY requirements.txt ./
-COPY src/ ./src/
-
-RUN pip3 install -r requirements.txt
-
-EXPOSE 8501
-
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
-
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+COPY . /app
+RUN pip install --no-cache-dir -r requirements.txt
+EXPOSE 5000
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,16 @@
+from flask import Flask, request, jsonify
+import pandas as pd
+import joblib
+
+app = Flask(__name__)
+model = joblib.load("xgb_model_pipeline.pkl")
+
+@app.route('/predict', methods=['POST'])
+def predict():
+    data = request.get_json(force=True)
+    input_df = pd.DataFrame([data])
+    prediction = model.predict(input_df)[0]
+    return jsonify({'conversion': int(prediction)})
+
+if __name__ == '__main__':
+    app.run(debug=True, host='0.0.0.0')

requirements.txt

@@ -1,3 +1,5 @@
-altair
+flask
 pandas
-streamlit
+scikit-learn
+xgboost
+joblib

xgb_model_pipeline.pkl

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