Update README.md

README.md

@@ -14,23 +14,22 @@ license: apache-2.0
 
 ```python
 # !pip install diffusers
+import torch
 from diffusers import DiffusionPipeline
 import PIL.Image
-import numpy as np
 
 model_id = "fusing/glide-base"
 
 # load model and scheduler
-ddpm = DiffusionPipeline.from_pretrained(model_id)
+pipeline = DiffusionPipeline.from_pretrained(model_id)
 
-# run pipeline in inference (sample random noise and denoise)
-image = ddpm(eta=0.0, num_inference_steps=50)
+# run inference (text-conditioned denoising + upscaling)
+img = pipeline("a corgi wearing a red bowtie and a purple party hat")
 
 # process image to PIL
-image_processed = image.cpu().permute(0, 2, 3, 1)
-image_processed = (image_processed + 1.0) * 127.5
-image_processed = image_processed.numpy().astype(np.uint8)
-image_pil = PIL.Image.fromarray(image_processed[0])
+img = img.squeeze(0)
+img = ((img + 1)*127.5).round().clamp(0, 255).to(torch.uint8).cpu().numpy()
+image_pil = PIL.Image.fromarray(img)
 
 # save image
 image_pil.save("test.png")
@@ -38,7 +37,6 @@ image_pil.save("test.png")
 
 ## Samples
 
-1. ![sample_1](https://huggingface.co/datasets/patrickvonplaten/images/resolve/main/hf/ddim-lsun-bedroom/image_0.png)
-2. ![sample_1](https://huggingface.co/datasets/patrickvonplaten/images/resolve/main/hf/ddim-lsun-bedroom/image_1.png)
-3. ![sample_1](https://huggingface.co/datasets/patrickvonplaten/images/resolve/main/hf/ddim-lsun-bedroom/image_2.png)
-4. ![sample_1](https://huggingface.co/datasets/patrickvonplaten/images/resolve/main/hf/ddim-lsun-bedroom/image_3.png)
+1. ![sample_1](https://huggingface.co/datasets/anton-l/images/resolve/main/glide1.png)
+2. ![sample_2](https://huggingface.co/datasets/anton-l/images/resolve/main/glide2.png)
+3. ![sample_3](https://huggingface.co/datasets/anton-l/images/resolve/main/glide3.png)