Update evo_nishikie_v1.py

.gitattributes

@@ -33,4 +33,3 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-test.jpg filter=lfs diff=lfs merge=lfs -text

README.md

@@ -33,36 +33,21 @@ Use the code below to get started with the model.
    ```
    git clone https://huggingface.co/SakanaAI/Evo-Nishikie-v1
    ```
-2. Install git-lfs if you don't have it yet.
-   ```
-   sudo apt install git-lfs
-   git lfs install
-   ```
-3. Create conda env
-   ```
-   conda create -n evo-nishikie python=3.11
-   conda activate evo-nishikie
-   ```
-4. Install packages
+2. Install packages
    ```
    cd Evo-Nishikie-v1
    pip install -r requirements.txt
    ```
-5. Run
+3. Run
    ```python
-   import torch
    from io import BytesIO
    from PIL import Image
    import requests
    from evo_nishikie_v1 import load_evo_nishikie
    
-   #Get image from URL
-   #url = "https://huggingface.co/spaces/SakanaAI/Evo-Nishikie/resolve/main/sample2.jpg"
-   #original_image = Image.open(BytesIO(requests.get(url).content))
-   
-   #Use local image
-   original_image = Image.open('test.jpg')
-   
+   url = "https://huggingface.co/spaces/SakanaAI/Evo-Nishikie/resolve/main/sample2.jpg"
+   original_image = Image.open(BytesIO(requests.get(url).content))
+
    # Generate
    device = "cuda"
    pipe, processor = load_evo_nishikie(device)
@@ -72,12 +57,12 @@ Use the code below to get started with the model.
         image=processor(original_image),
         guidance_scale=7.0,
         controlnet_conditioning_scale=0.8,
-        num_inference_steps=35,
+        num_inference_steps=50,
+        generator=torch.Generator(device).manual_seed(0),
         num_images_per_prompt=1,
         output_type="pil",
    ).images
    images[0].save("out.png")
-
    ```
 
 </details>

requirements.txt

@@ -4,6 +4,7 @@ torchvision
 accelerate==0.32.0
 controlnet-aux==0.0.9
 diffusers==0.29.2
+gradio==4.38.1
 sentencepiece==0.2.0
 transformers==4.42.3
 peft==0.11.1