diff --git a/deployment/handlers/prediction_app.py b/deployment/handlers/prediction_app.py
index 7555934..9d98882 100644
--- a/deployment/handlers/prediction_app.py
+++ b/deployment/handlers/prediction_app.py
@@ -87,11 +87,19 @@ def handler(event, context):
             import pandas as pd
 
             model.load_model(model_filepath)
+
+            warmup_df = pd.DataFrame(columns=model.model.original_features)
+            warmup_df = pd.concat(
+                [warmup_df.T, pd.DataFrame([0] * len(warmup_df.T))], axis=1
+            ).T
+            warmup_df.fillna(0, inplace=True)
+
+            model.predict(data=warmup_df)
             return {
                 "statusCode": 200,
                 "body": json.dumps(
                     {
-                        "message": f"{model.predict(data=pd.DataFrame(columns=model.model.original_features))}"
+                        "message": "Successfully processed warm up invocation",
                     }
                 ),
             }