klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 21:19:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve google image generation configuration

Browse Source
This commit is contained in:
Debanjum
2025-06-20 00:36:55 -07:00
parent dc202e4441
commit 4448ab665c
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/khoj/processor/image/generate.py
View File

@@ -242,7 +242,13 @@ def generate_image_with_google(
client = genai.Client(api_key=api_key) client = genai.Client(api_key=api_key)
# Configure image generation settings # Configure image generation settings
config = gtypes.GenerateImagesConfig(number_of_images=1) config = gtypes.GenerateImagesConfig(
number_of_images=1,
safety_filter_level=gtypes.SafetyFilterLevel.BLOCK_LOW_AND_ABOVE,
person_generation=gtypes.PersonGeneration.ALLOW_ADULT,
include_rai_reason=True,
output_mime_type="image/png",
)
# Call the Gemini API to generate the image # Call the Gemini API to generate the image
response = client.models.generate_images(model=text2image_model, prompt=improved_image_prompt, config=config) response = client.models.generate_images(model=text2image_model, prompt=improved_image_prompt, config=config)