Update README.md

README.md

@@ -19,15 +19,15 @@ widget:
 - text: transform the setting to a winter scene
   output:
     url: >-
-      https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/green_creature.jpg
+      https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_green_creature.jpg
 - text: turn the color of mushroom to gray
   output:
     url: >-
-      https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/mushroom.jpg
+      https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_mushroom.jpg
 - text: Change it to look like it's in the style of an impasto painting.
   output:
     url: >-
-      https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/norte_dam.jpg
+      https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_norte_dam.jpg
 ---
 
 # Flux Edit
@@ -53,13 +53,14 @@ from diffusers import FluxControlPipeline, FluxTransformer2DModel
 from diffusers.utils import load_image
 import torch 
 
-path = "sayakpaul/FLUX.1-dev-edit-v0" # to change
+path = "sayakpaul/FLUX.1-dev-edit-v0" 
 edit_transformer = FluxTransformer2DModel.from_pretrained(path, torch_dtype=torch.bfloat16)
 pipeline = FluxControlPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev", transformer=edit_transformer, torch_dtype=torch.bfloat16
 ).to("cuda")
 
-image = load_image("./assets/mushroom.jpg") # resize as needed.
+url = "https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/assets/mushroom.jpg"
+image = load_image(url) # resize as needed.
 print(image.size)
 
 prompt = "turn the color of mushroom to gray"
@@ -91,7 +92,7 @@ from diffusers.utils import load_image
 from huggingface_hub import hf_hub_download
 import torch
 
-path = "sayakpaul/FLUX.1-dev-edit-v0" # to change
+path = "sayakpaul/FLUX.1-dev-edit-v0"
 edit_transformer = FluxTransformer2DModel.from_pretrained(path, torch_dtype=torch.bfloat16)
 pipeline = FluxControlPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev", transformer=edit_transformer, torch_dtype=torch.bfloat16
@@ -103,7 +104,9 @@ pipeline.load_lora_weights(
 )
 pipeline.set_adapters(["hyper-sd"], adapter_weights=[0.125])
 
-image = load_image("./assets/mushroom.jpg") # resize as needed.
+
+url = "https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/assets/mushroom.jpg"
+image = load_image(url) # resize as needed.
 print(image.size)
 
 prompt = "turn the color of mushroom to gray"