alanchen1115 commited on
Commit
922ec13
·
verified ·
1 Parent(s): b02c30c

Update main.py

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