Update README.md

README.md

@@ -22,6 +22,7 @@ The diffuser generate a forward moving path of a panoramic input image, to explo
 ```python
 from diffusers import UNetSpatioTemporalConditionModel, StableVideoDiffusionPipeline
 import torch
+from PIL import Image
 
 model_id = 'TaiMingLu/GenEx-World-Explorer'
 
@@ -41,6 +42,17 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(
     torch_dtype=torch.float16,
     local_files_only=True,
 ).to('cuda')
+
+# Explore the world!
+image = Image.open('example.png').resize((1024, 576), Image.BICUBIC).convert('RGB')
+
+generator = torch.manual_seed(-1)
+with torch.inference_mode():
+    frames = self.pipe(image,
+                num_frames=25,
+                width=1024,
+                height=576,
+                decode_chunk_size=8, generator=generator, motion_bucket_id=127, fps=7, num_inference_steps=30, noise_aug_strength=0.02).frames[0]
 ```
 
 ## 🔧 Requirements