alanchen1115 commited on
Commit
64f1423
·
verified ·
1 Parent(s): f647f26

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -104,7 +104,7 @@ def generate_and_upload_image(prompt: str) -> str:
104
  """
105
  try:
106
  response = genai_client.models.generate_content(
107
- model="gemini-2.5-flash-image-preview",
108
  contents=prompt,
109
  config=types.GenerateContentConfig(response_modalities=['Text', 'Image'])
110
  )
 
104
  """
105
  try:
106
  response = genai_client.models.generate_content(
107
+ model="gemini-2.0-flash-preview-image-generation",
108
  contents=prompt,
109
  config=types.GenerateContentConfig(response_modalities=['Text', 'Image'])
110
  )