Update README.md

README.md

@@ -1 +1,16 @@
-This is a dummy model used for testing `NeuronStableDiffusionPipeline` of `optimum-neuron`.
+This is a dummy model used for testing `NeuronStableDiffusionPipeline` of `optimum-neuron`.
+
+```python
+from optimum.neuron import NeuronStableDiffusionPipeline
+
+model_id = "optimum/tiny-stable-diffusion-neuronx"
+input_shapes = {"batch_size": 2, "sequence_length": 18, "num_channels": 4, "height": 64, "width": 64}
+
+stable_diffusion = NeuronStableDiffusionPipeline.from_pretrained(model_id, export=True, **input_shapes)
+save_directory = "sd_neuron/"
+stable_diffusion.save_pretrained(save_directory)
+
+
+prompt = "sailing ship in storm by Leonardo da Vinci"
+image = stable_diffusion(prompt).images[0]
+```