Upload README.md with huggingface_hub

README.md

@@ -33,36 +33,42 @@ Here are some examples of images generated using this style LoRA:
 
 ## Inference Example
 ```python
-from huggingface_hub import hf_hub_download
 from diffusers import FluxKontextPipeline
 from diffusers.utils import load_image
 import torch
 
-# Define the style and model details
-STYLE_NAME = "Snoopy"
-LORA_FILENAME = "Snoopy_lora_weights.safetensors"
-REPO_ID = "Kontext-Style/Snoopy_lora"
+# Load the base pipeline
+pipeline = FluxKontextPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-Kontext-dev", 
+    torch_dtype=torch.bfloat16
+).to('cuda')
 
-# Download the LoRA weights
-# Make sure you have created a folder named 'LoRAs' in your current directory
-hf_hub_download(repo_id=REPO_ID, filename=LORA_FILENAME, local_dir="./LoRAs")
+# Load the LoRA adapter for the Snoopy style directly from the Hub
+pipeline.load_lora_weights("Kontext-Style/Snoopy_lora", weight_name="Snoopy_lora_weights.safetensors", adapter_name="lora")
+pipeline.set_adapters(["lora"], adapter_weights=[1])
 
-# Load an image
+# Load a source image (you can use any image)
 image = load_image("https://huggingface.co/datasets/black-forest-labs/kontext-bench/resolve/main/test/images/0003.jpg").resize((1024, 1024))
 
-# Load the pipeline
-pipeline = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to('cuda')
-
-# Load and set the LoRA adapter
-pipeline.load_lora_weights(f"./LoRAs/{LORA_FILENAME}", adapter_name="lora")
-pipeline.set_adapters(["lora"], adapter_weights=[1])
+# Prepare the prompt
+# The style_name is used in the prompt and for the output filename.
+style_name = "Snoopy"
+prompt = f"Turn this image into the Snoopy style."
 
 # Run inference
-prompt = f"Turn this image into the {STYLE_NAME.replace('_', ' ')} style."
-result_image = pipeline(image=image, prompt=prompt, height=1024, width=1024, num_inference_steps=24).images[0]
-result_image.save(f"{STYLE_NAME}.png")
+result_image = pipeline(
+    image=image, 
+    prompt=prompt, 
+    height=1024, 
+    width=1024, 
+    num_inference_steps=24
+).images[0]
+
+# Save the result
+output_filename = f"{style_name.replace(' ', '_')}.png"
+result_image.save(output_filename)
 
-print(f"Image saved as {STYLE_NAME}.png")
+print(f"Image saved as {output_filename}")
 ```
 
 Feel free to open an issue or contact us for feedback or collaboration!