Update README.md

README.md

@@ -70,6 +70,7 @@ from diffusers import ControlNetModel, StableDiffusionXLControlNetPipeline
 from controlnet_aux import OpenposeDetector
 import torch
 from diffusers.utils import load_image
+from PIL import Image
 
 controlnet = ControlNetModel.from_pretrained(
     "briaai/BRIA-2.3-ControlNet-Pose",
@@ -83,15 +84,16 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
 )
 pipe.to("cuda")
 
-prompt = "A portrait of a Beautiful and playful ethereal singer, golden designs, highly detailed, blurry background"
+prompt = "Two kids in bright orange jackets play near a blue tent in a forest with silver-leafed trees.,photography"
 negative_prompt = "Logo,Watermark,Text,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate,Mutilated,Mutilated hands,Poorly drawn face,Deformed,Bad anatomy,Cloned face,Malformed limbs,Missing legs,Too many fingers"
 
 # Calculate Pose image
 openpose = OpenposeDetector.from_pretrained('lllyasviel/ControlNet')
 
-image = load_image("https://huggingface.co/briaai/BRIA-2.3-ControlNet-Pose/resolve/main/test_image.png")
+image = load_image("https://huggingface.co/briaai/BRIA-2.3-ControlNet-Pose/resolve/main/test_image.jpg")
 
-pose_image = openpose(image)
+pose_image = openpose(image, include_body=True, include_hand=True, include_face=True)[0]
 
 image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=pose_image, controlnet_conditioning_scale=1.0, height=1024, width=1024).images[0]
+
 ```