diff --git a/deployment/serverless.yml b/deployment/serverless.yml
index b23158d..009f7db 100644
--- a/deployment/serverless.yml
+++ b/deployment/serverless.yml
@@ -51,3 +51,4 @@ functions:
           path: /predict
           method: POST
     timeout: 120 # Set max run time to 2 minutes - we shouldn't need this much time so this can be reviewed
+    memorySize: 4096