Update README.md

README.md

@@ -1,6 +1,23 @@
-## Gemini Prompt Expander
+# Gemini Prompt Expander
 
-```py
+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(
@@ -10,5 +27,12 @@ gemini_block = ModularPipelineBlocks.from_pretrained(
 gemini = gemini_block.init_pipeline()
 output = gemini(prompt="a dog sitting by the river, watching the sunset")
 print(f"{output.values['prompt']=}")
+```
+
+## Use in Mellon
+
+This block includes a `mellon_pipeline_config.json` for use with [Mellon](https://github.com/cubiq/Mellon):
 
-```
+1. Drag a **Dynamic Block Node** from the ModularDiffusers section
+2. Enter `diffusers-internal-dev/gemini-prompt-expander` as the repo_id
+3. The node will transform to show the prompt input and expanded prompt output