Update app.py
Browse files
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
|
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 |
|