Update README.md

README.md

@@ -13,13 +13,9 @@ library_name: diffusers
 ```
 from optimum.neuron import NeuronStableDiffusionXLPipeline
 
-model_id = "dminhk/stable-diffusion-xl-base-1.0-neuronx"
-input_shapes = {"batch_size": 1, "height": 1024, "width": 1024}
-
-stable_diffusion_xl = NeuronStableDiffusionXLPipeline.from_pretrained(model_id, export=False, **input_shapes, device_ids=[0,1])
-```
-
-```
 prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+stable_diffusion_xl = NeuronStableDiffusionXLPipeline.from_pretrained(
+    model_id="dminhk/stable-diffusion-xl-base-1.0-neuronx", device_ids=[0, 1]
+)
 image = stable_diffusion_xl(prompt).images[0]
 ```