Skip to content
This repository was archived by the owner on Jun 12, 2024. It is now read-only.

Commit 8be2c0f

Browse files
authored
fix: image generation
1 parent 0bab1e7 commit 8be2c0f

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

README.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -156,8 +156,8 @@ generated_images = response.generated_images # Check generated images [Dict]
156156
GeminiImage.save_sync(generated_images, save_path="cached_folder")
157157

158158
# You can use byte type image dict as follow:
159-
# bytes_images_dict = GeminiImage.fetch_bytes_sync(generated_images) # Get bytes images dict
160-
# GeminiImage.save_images_sync(bytes_images_dict, path="save_path") # Save to path
159+
# bytes_images_dict = GeminiImage.fetch_images_dict_sync(generated_images, cookies) # Get bytes images dict
160+
# GeminiImage.save_images_sync(bytes_images_dict, path="cached_folder") # Save to path
161161
```
162162
*Async*
163163
```python
@@ -187,7 +187,7 @@ async def fetch_and_save_images_async(prompt: str):
187187
response = await GeminiClient.generate_content_async(prompt)
188188

189189
generated_images = response.generated_images # Check response images [Dict]
190-
bytes_images_dict = await GeminiImage.fetch_images_dict(generated_images) # Get bytes images dict asynchronously
190+
bytes_images_dict = await GeminiImage.fetch_images_dict(generated_images, cookies) # Get bytes images dict asynchronously
191191
await GeminiImage.save_images(bytes_images_dict, path="cached_folder") # Save to path asynchronously
192192

193193
# Run the async function
@@ -216,7 +216,7 @@ response_images = response.web_images # Check response images [Dict]
216216
GeminiImage.save_sync(response_images, save_path="cached_folder")
217217

218218
# You can use byte type image dict as follow:
219-
# bytes_images_dict = GeminiImage.fetch_bytes_sync(response_images) # Get bytes images dict
219+
# bytes_images_dict = GeminiImage.fetch_bytes_sync(response_images, cookies) # Get bytes images dict
220220
# GeminiImage.save_images_sync(bytes_images_dict, path="cached_folder") # Save to path
221221
```
222222
*Async*
@@ -247,7 +247,7 @@ async def fetch_and_save_images_async(prompt: str):
247247
response = await GeminiClient.generate_content_async(prompt)
248248

249249
response_images = response.web_images # Check response images [Dict]
250-
bytes_images_dict = await GeminiImage.fetch_images_dict(response_images) # Get bytes images dict asynchronously
250+
bytes_images_dict = await GeminiImage.fetch_images_dict(response_images, cookies) # Get bytes images dict asynchronously
251251
await GeminiImage.save_images(bytes_images_dict, path="cached_folder") # Save to path asynchronously
252252

253253
# Run the async function

0 commit comments

Comments
 (0)