Update README.md

README.md

@@ -2,7 +2,6 @@
 license: apache-2.0
 language:
 - en
-- zh
 library_name: diffusers
 pipeline_tag: image-to-image
 ---
@@ -16,39 +15,31 @@ pipeline_tag: image-to-image
 # Usage
 
 ```python
-import os
 import torch
-from PIL import Image
-from diffusers import QwenImageEditPlusPipeline
+from diffusers import FluxKontextPipeline
+from diffusers.utils import load_image
 
-pipeline = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509", torch_dtype=torch.bfloat16)
-print("pipeline loaded")
+pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16)
 
-pipeline.load_lora_weights(
+pipe.load_lora_weights(
     "chestnutlzj/Edit-R1-Qwen-Image-Edit-2509",
     adapter_name="lora",
 )
-pipeline.set_adapters(["lora"], adapter_weights=[1])
-
-pipeline.to('cuda')
-pipeline.set_progress_bar_config(disable=None)
-image1 = Image.open("input1.png")
-image2 = Image.open("input2.png")
-prompt = "The magician bear is on the left, the alchemist bear is on the right, facing each other in the central park square."
-inputs = {
-    "image": [image1, image2],
-    "prompt": prompt,
-    "generator": torch.manual_seed(0),
-    "true_cfg_scale": 4.0,
-    "negative_prompt": " ",
-    "num_inference_steps": 40,
-    "guidance_scale": 1.0,
-    "num_images_per_prompt": 1,
-}
-with torch.inference_mode():
-    output = pipeline(**inputs)
-    output_image = output.images[0]
-    output_image.save("output_image_edit_plus.png")
-    print("image saved at", os.path.abspath("output_image_edit_plus.png"))
+pipe.set_adapters(["lora"], adapter_weights=[1])
+
+pipe.to("cuda")
+
+
+input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
+
+image = pipe(
+  image=input_image,
+  prompt="Add a hat to the cat",
+  guidance_scale=2.5
+).images[0]
 
 ```
+
+# Licence
+
+FLUX.1-Kontext-dev falls under the [FLUX.1 [dev] Non-Commercial License](https://github.com/black-forest-labs/flux/blob/main/model_licenses/LICENSE-FLUX1-dev).