Update app.py

app.py

@@ -43,3 +43,27 @@ def main():
 
 if __name__ == "__main__":
     main()
+import onnxruntime as ort
+import gradio as gr
+import numpy as np
+
+sess = ort.InferenceSession("collusion_xgb_model.onnx")
+
+def predict(amount, user_txn, driver_txn, pair_count, hour, day):
+    X = np.array([[amount, user_txn, driver_txn, pair_count, hour, day]], dtype=np.float32)
+    pred = sess.run(None, {sess.get_inputs()[0].name: X})[0][0][0]
+    return float(pred)
+
+gr.Interface(
+    fn=predict,
+    inputs=[
+        gr.Number(label="Amount"),
+        gr.Number(label="User Txn Count"),
+        gr.Number(label="Driver Txn Count"),
+        gr.Number(label="User-Driver Pair Count"),
+        gr.Number(label="Hour"),
+        gr.Number(label="Day of Week"),
+    ],
+    outputs=gr.Number(label="Fraud Probability"),
+    title="Collusion Fraud Detection"
+).launch()