rename INPUT_TEXT environment variable to INPUT

DevlinRocha · Jan 29, 2025 · 00dddeb · 00dddeb
1 parent 1e20094
commit 00dddeb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/create_lilypad_module/templates/lilypad_module.json.tmpl b/src/create_lilypad_module/templates/lilypad_module.json.tmpl
@@ -16,7 +16,7 @@
                 "Entrypoint": ["python", "/workspace/src/run_inference.py"],
                 "WorkingDirectory": "/workspace",
                 "EnvironmentVariables": [
-                    "INPUT_TEXT={{ js .input }}"
+                    "INPUT={{ js .input }}"
                 ],
                 "Image": "",
             },

diff --git a/src/create_lilypad_module/templates/scripts/run_module.py b/src/create_lilypad_module/templates/scripts/run_module.py
@@ -32,7 +32,7 @@ def run_module():
             "docker",
             "run",
             "-e",
-            f"INPUT_TEXT={input}",
+            f"INPUT={input}",
             "-v",
             "$(pwd)/outputs:/outputs",
             f"{DOCKER_REPO}:latest",