userIdc2024 commited on
Commit
e799db8
·
verified ·
1 Parent(s): cfabdf9

Update helpercodes/image_functions.py

Browse files
Files changed (1) hide show
  1. helpercodes/image_functions.py +1 -1
helpercodes/image_functions.py CHANGED
@@ -165,7 +165,7 @@ def generate_image(prompt: str, aspect_ratio='1:1', design="None") -> Image.Imag
165
  "magic_prompt_option": "Off",
166
  "style_type": design
167
  }
168
- output = replicate_client.run("ideogram-ai/ideogram-v2-turbo", input=inputs)
169
 
170
  if not output:
171
  logging.error("API response is empty. Image generation failed.")
 
165
  "magic_prompt_option": "Off",
166
  "style_type": design
167
  }
168
+ output = replicate_client.run("ideogram-ai/ideogram-v2a", input=inputs)
169
 
170
  if not output:
171
  logging.error("API response is empty. Image generation failed.")