feat: add custom layer

petercat-ai · Apr 15, 2024 · e5fc349 · e5fc349
1 parent fe528f0
commit e5fc349
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/server/main.py b/server/main.py
@@ -15,6 +15,10 @@
 
 open_api_key = get_env_variable("OPENAI_API_KEY")
 is_dev = bool(get_env_variable("IS_DEV"))
+task_root = get_env_variable("LAMBDA_TASK_ROOT")
+
+if task_root:
+    os.environ["NLTK_DATA"] = os.path.join(task_root, "nltk_data")
 
 app = FastAPI( 
     title="Bo-meta Server",