add init image

app_tensorrt.py

@@ -7,6 +7,7 @@ from diffusers import (
     DDPMScheduler,
     StableDiffusionXLImg2ImgPipeline,
     AutoencoderKL,
+    load_image,
 )
 
 os.system("pip install torch_tensorrt==2.4.0")
@@ -48,8 +49,9 @@ compiledModel = torch.compile(
 
 base_pipe.unet = compiledModel
 
+init_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/img2img.png")
 generated_image = base_pipe(
-    prompt="A beautiful sunset over the city",
+    init_image,
     num_inference_steps=5,
 ).images[0]