From d7370248a220820cbcba154626f6e1842cd5e3a5 Mon Sep 17 00:00:00 2001
From: Michael Duong <michaelduong22@gmail.com>
Date: Mon, 3 Nov 2025 22:40:35 +0000
Subject: [PATCH] use model columns as data

---
 deployment/handlers/prediction_app.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/deployment/handlers/prediction_app.py b/deployment/handlers/prediction_app.py
index 7deae3a..7555934 100644
--- a/deployment/handlers/prediction_app.py
+++ b/deployment/handlers/prediction_app.py
@@ -90,7 +90,9 @@ def handler(event, context):
             return {
                 "statusCode": 200,
                 "body": json.dumps(
-                    {"message": f"{model.predict(data=pd.DataFrame({'a': [1]}))}"}
+                    {
+                        "message": f"{model.predict(data=pd.DataFrame(columns=model.model.original_features))}"
+                    }
                 ),
             }
         else: