| # Gemini Prompt Expander |
|
|
| A custom ModularPipelineBlocks that uses Google's Gemini API to expand short prompts into detailed, vivid image generation prompts. |
|
|
| ## Requirement |
|
|
| Install the Google Generative AI package: |
| ```bash |
| pip install google-generativeai |
| ``` |
|
|
| ## Setup |
|
|
| Get your Gemini API key from [Google AI Studio](https://aistudio.google.com/apikey) and set it as an environment variable: |
| ```bash |
| export GOOGLE_API_KEY="your-api-key-here" |
| ``` |
|
|
| ## Usage |
| ```python |
| from diffusers.modular_pipelines import ModularPipelineBlocks |
| |
| gemini_block = ModularPipelineBlocks.from_pretrained( |
| "diffusers-internal-dev/gemini-prompt-expander", |
| trust_remote_code=True, |
| ) |
| gemini = gemini_block.init_pipeline() |
| output = gemini(prompt="a dog sitting by the river, watching the sunset") |
| print(f"{output.values['prompt']=}") |
| ``` |
