Update README.md

README.md

@@ -41,13 +41,16 @@ Use the code below to get started with the model.
 3. Run
    ```python
    from io import BytesIO
+   from PIL import Image
+   import requests
    from evo_nishikie_v1 import load_evo_nishikie
    
    url = "https://huggingface.co/spaces/SakanaAI/Evo-Nishikie/resolve/main/sample2.jpg"
    original_image = Image.open(BytesIO(requests.get(url).content))
 
    # Generate
-   pipe, processor = load_evo_nishikie()
+   device = "cuda"
+   pipe, processor = load_evo_nishikie(device)
    images = pipe(
         prompt="着物を着た女性が、赤ん坊を抱え、もう一人の子どもが手押し車を引いています。背景には木があります。最高品質の輻の浮世絵。超詳細。",
         negative_prompt="暗い",
@@ -55,7 +58,7 @@ Use the code below to get started with the model.
         guidance_scale=7.0,
         controlnet_conditioning_scale=0.8,
         num_inference_steps=50,
-        generator=torch.Generator().manual_seed(0),
+        generator=torch.Generator(device).manual_seed(0),
         num_images_per_prompt=1,
         output_type="pil",
    ).images