Update README.md

README.md

@@ -23,19 +23,17 @@ Please raise any concerns you may have.
 ## Usage
 
 ```python
-# !pip install diffusers
-from diffusers import LDMTextToImagePipeline
-import PIL.Image
-import numpy as np
+# !pip install diffusers transformers
+from diffusers import DiffusionPipeline
 
 model_id = "CompVis/ldm-text2im-large-256"
 
 # load model and scheduler
-ldm = LDMTextToImagePipeline.from_pretrained(model_id)
+ldm = DiffusionPipeline.from_pretrained(model_id)
 
 # run pipeline in inference (sample random noise and denoise)
 prompt = "A painting of a squirrel eating a burger"
-images = ldm([prompt], num_inference_steps=50, eta=0.3, guidance_scale=0.6, batch_size=1)["sample"]
+images = ldm([prompt], num_inference_steps=50, eta=0.3, guidance_scale=6)["sample"]
 
 # save images
 for idx, image in enumerate(images):
@@ -43,4 +41,12 @@ for idx, image in enumerate(images):
 ```
 
 ## Demo
-[Hugging Face Spaces](https://huggingface.co/spaces/CompVis/ldm-text2im-large-256-diffusers)
+
+[Hugging Face Spaces](https://huggingface.co/spaces/CompVis/ldm-text2im-large-256-diffusers)
+
+## Samples
+
+1. ![sample_0](https://huggingface.co/CompVis/ldm-text2im-large-256/resolve/main/images/squirrel-0.png)
+2. ![sample_1](https://huggingface.co/CompVis/ldm-text2im-large-256/resolve/main/images/squirrel-1.png)
+3. ![sample_2](https://huggingface.co/CompVis/ldm-text2im-large-256/resolve/main/images/squirrel-2.png)
+4. ![sample_3](https://huggingface.co/CompVis/ldm-text2im-large-256/resolve/main/images/squirrel-3.png)