Upload 3 files

app.py

@@ -0,0 +1,24 @@
+from flask import Flask, request, jsonify
+import joblib
+import numpy as np
+
+# Load trained LightGBM Model
+model = joblib.load("Lightgbm_model.pkl")
+
+app = Flask(__name__)
+
+@app.route('/predict', methods=['POST'])
+def predict():
+  data = request.json
+  features = np.array(data['features']).reshape(1, -1)
+
+  # Get probability prediction
+  probs = model.predict_proba(features)[:,1]
+
+  # Apply threshold 0.3
+  prediction = init(probs[0] > 0.3)
+
+  return jsonify({'attack_detected': prediction})
+
+if __name__ == '__main__':
+  app.run(host='0.0.0.0', port=5000)

lightgbm_model.pkl

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

requirements.txt

@@ -0,0 +1,4 @@
+flask
+koblib
+numpy
+scikit-learn