Noo88ear commited on
Commit
ac520ab
·
verified ·
1 Parent(s): 8ff3020

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -172,13 +172,15 @@ def generate_marketing_image(prompt: str, style: str = "realistic") -> str:
172
  # Initialize the genai SDK client
173
  client = genai_sdk.Client(api_key=GOOGLE_API_KEY)
174
 
175
- # Generate image using Imagen 4.0 (best balance of quality and speed)
176
  result = client.models.generate_images(
177
  model="imagen-4.0-generate-preview-06-06",
178
  prompt=enhanced_prompt,
179
  config={
180
  "number_of_images": 1,
181
- "output_mime_type": "image/png"
 
 
182
  }
183
  )
184
 
 
172
  # Initialize the genai SDK client
173
  client = genai_sdk.Client(api_key=GOOGLE_API_KEY)
174
 
175
+ # Generate image using Imagen 4.0 with reduced safety filtering
176
  result = client.models.generate_images(
177
  model="imagen-4.0-generate-preview-06-06",
178
  prompt=enhanced_prompt,
179
  config={
180
  "number_of_images": 1,
181
+ "output_mime_type": "image/png",
182
+ "safety_filter_level": "block_low_and_above",
183
+ "include_safety_attributes": False
184
  }
185
  )
186