diff --git a/3d-generation-pipeline/.env.example b/3d-generation-pipeline/.env.example
index ca034486..6da7eace 100644
--- a/3d-generation-pipeline/.env.example
+++ b/3d-generation-pipeline/.env.example
@@ -1,4 +1,5 @@
 INVOKEAI_URL=
+INVOKEAI_MODEL_KEY=
 TRELLIS_URL=
 
 CLOUDFLARE_ACCOUNT_ID=
diff --git a/3d-generation-pipeline/generate_image_local.py b/3d-generation-pipeline/generate_image_local.py
index 2a7c4dd8..18c31226 100644
--- a/3d-generation-pipeline/generate_image_local.py
+++ b/3d-generation-pipeline/generate_image_local.py
@@ -18,6 +18,7 @@ logger = logging.getLogger("invokeai-mcp")
 load_dotenv()
 
 INVOKEAI_BASE_URL = os.environ["INVOKEAI_URL"]
+INVOKEAI_MODEL_KEY = os.environ["INVOKEAI_MODEL_KEY"]
 DEFAULT_QUEUE_ID = "default"
 
 # HTTP client
@@ -34,13 +35,12 @@ def get_client() -> httpx.AsyncClient:
 
 
 async def text_to_image_invoke_ai(prompt, output_path):
-    # see available model keys via GET http://127.0.0.1:9090/api/v2/models/?model_type=main
+    # see available model keys via GET http://INVOKEAI_BASE_URL:9090/api/v2/models/?model_type=main
     args = {
         "prompt": prompt,
         "width": 512,
         "height": 512,
-        "model_key": "79401292-0a6b-428d-b7d7-f1e86caeba2b" # Juggernaut XL v9
-        #"model_key": "735f6485-6703-498f-929e-07cf0bbbd179" # Dreamshaper 8
+        "model_key": INVOKEAI_MODEL_KEY
     }
     image_url = await generate_image(args)
     print("got image url: ", image_url)