From cf88959a2fa51b74e105dfd836b62bca7b9061d8 Mon Sep 17 00:00:00 2001
From: Katie Nguyen <21978337+katiemn@users.noreply.github.com>
Date: Thu, 14 Aug 2025 10:16:35 -0700
Subject: [PATCH] feat(genai): update Imagen 4 model

---
 genai/image_generation/imggen_with_txt.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/genai/image_generation/imggen_with_txt.py b/genai/image_generation/imggen_with_txt.py
index eb1b984ec9..92f88e675b 100644
--- a/genai/image_generation/imggen_with_txt.py
+++ b/genai/image_generation/imggen_with_txt.py
@@ -26,7 +26,7 @@ def generate_images(output_file: str) -> Image:
     # output_file = "output-image.png"
 
     image = client.models.generate_images(
-        model="imagen-4.0-generate-preview-06-06",
+        model="imagen-4.0-generate-001",
         prompt="A dog reading a newspaper",
         config=GenerateImagesConfig(
             image_size="2K",