Fix example

README.md

@@ -16,18 +16,21 @@ This repo contains ONNX model files for [madebyollin's Tiny AutoEncoder for Stab
 To install the requirements for this demo, do pip install optimum["onnxruntime"]. 
 
 ```python
+from huggingface_hub import snapshot_download
+from diffusers.pipelines import OnnxRuntimeModel
 from optimum.onnxruntime import ORTStableDiffusionPipeline
 
 model_id = "CompVis/stable-diffusion-v1-4"
-pipeline = ORTStableDiffusionPipeline.from_pretrained(model_id, revision="onnx")
 
-# Inject TAESD
+taesd_dir = snapshot_download(repo_id="deinferno/taesd-onnx")
 
-taesd_id = "deinferno/taesd-onnx"
-pipeline.vae_decoder = OnnxRuntimeModel.from_pretrained(taesd_id, subfolder = "vae_decoder")
-pipeline.vae_encoder = OnnxRuntimeModel.from_pretrained(taesd_id, subfolder = "vae_encoder")
+pipeline = ORTStableDiffusionPipeline.from_pretrained(
+        model_id,
+        vae_decoder_session = OnnxRuntimeModel.from_pretrained(f"{taesd_dir}/vae_decoder"),
+        vae_encoder_session = OnnxRuntimeModel.from_pretrained(f"{taesd_dir}/vae_encoder"),
+        revision="onnx")
 
 prompt = "sailing ship in storm by Leonardo da Vinci"
 image = pipeline(prompt).images[0]
-output.save("result.png")
+image.save("result.png")
 ```