Update README.md

README.md

@@ -56,16 +56,16 @@ from diffusers import DiffusionPipeline
 import torch
 
 pipe = DiffusionPipeline.from_pretrained(
-    "playgroundai/playground-v2-1024px-aesthetic",
+    "playgroundai/playground-v2-512px-base",
     torch_dtype=torch.float16,
     use_safetensors=True,
     add_watermarker=False,
-    variant="fp16"
+    variant="fp16",
 )
 pipe.to("cuda")
 
 prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
-image  = pipe(prompt=prompt).images[0]
+image = pipe(prompt=prompt, width=512, height=512).images[0]
 ```
 
 ### User Study