Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -105,7 +105,7 @@ def generate_and_upload_image(prompt: str) -> str:
|
|
| 105 |
response = genai_client.models.generate_content(
|
| 106 |
model="gemini-2.5-flash-image",
|
| 107 |
contents=prompt,
|
| 108 |
-
|
| 109 |
)
|
| 110 |
|
| 111 |
image_binary = None
|
|
|
|
| 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 |
)
|
| 110 |
|
| 111 |
image_binary = None
|