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 |
|