oluinioluwa814 commited on
Commit
aba02c8
·
verified ·
1 Parent(s): f1ddd45

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -16,7 +16,7 @@ def generate_image(prompt: str, style: str):
16
  )
17
 
18
  result = client.models.generate_images(
19
- model="gemini-3-pro-image-preveiw",
20
  prompt=full_prompt,
21
  config=genai.types.GenerateImagesConfig(
22
  number_of_images=1,
 
16
  )
17
 
18
  result = client.models.generate_images(
19
+ model="imagen-3.0-generate-002",
20
  prompt=full_prompt,
21
  config=genai.types.GenerateImagesConfig(
22
  number_of_images=1,