Update README.md

README.md

@@ -7,7 +7,7 @@ library_name: diffusers
 This contains mellon config for A single-node Mellon block for the FLUX.2-klein-4B pipeline. 
 
 
-## How to create a modular repo?
+## How to create a modular repo like this one?
 
 load the pipeline and save it to a new Hub repository:
 ```python
@@ -29,16 +29,18 @@ from diffusers.modular_pipelines.mellon_node_utils import MellonPipelineConfig
 
 mellon_config = MellonPipelineConfig.from_custom_block(
     pipe.blocks,
-    mellon_types={
+    input_types={
         "prompt": "textbox",
         "image": "image",
-        "images": "image",
         "num_inference_steps": "number",
+    },
+    output_types={
+        "images": "image"
     }
 )
 
 mellon_config.save(
-    local_dir="local/dir",
+    "local/dir",
     repo_id="your-username/flux2-klein-auto-blocks",
     push_to_hub=True
 )
@@ -57,7 +59,7 @@ The generated `mellon_pipeline_config.json` may need adjustments. You can:
 ## Use in Mellon
 
 1. Drag a **Dynamic Block Node** from the ModularDiffusers section
-2. Enter `your-username/flux2-klein-auto-blocks` as the repo_id
+2. Enter `diffusers/FLUX.2-klein-4B-modular` and click "LOAD CUSTOM BLOCK" 
 3. The node will show your configured inputs (prompt, image, num_inference_steps) and outputs (images)
 
 ## Supported Mellon Types