diff --git a/3d-generation-pipeline/generate_image_local.py b/3d-generation-pipeline/generate_image_local.py
index de8a5193..c32f52ac 100644
--- a/3d-generation-pipeline/generate_image_local.py
+++ b/3d-generation-pipeline/generate_image_local.py
@@ -73,8 +73,11 @@ def download_file(url, filepath):
 
 
 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
     args = {
-        "prompt": prompt
+        "prompt": prompt,
+        "model_key": "79401292-0a6b-428d-b7d7-f1e86caeba2b" # Juggernaut XL v9
+        #"model_key": "735f6485-6703-498f-929e-07cf0bbbd179" # Dreamshaper 8
     }
     image_url = await generate_image(args)
     print("got image url: ", image_url)