Update README.md

README.md

@@ -1,49 +1,50 @@
----
-library_name: diffusers
-pipeline_tag: video-to-video
----
-
-# GenEx-World-Initializer 🚀🌍
-
-**GenEx World Explorer** is a video generation pipeline built on top of [Stable Video Diffusion (SVD)](https://huggingface.co/stabilityai/stable-video-diffusion-img2vid-xt-1-1).
-. It takes a keyframe, and generates a temporally consistent video. This explorer version builds on SVD with a custom `UNetSpatioTemporalConditionModel`.
-
-The diffuser generate a forward moving path of a panoramic input image, to explore a given scene.
-
-
-![](examples.png)
-
-## 📦 Usage
-
-```python
-from diffusers import UNetSpatioTemporalConditionModel, StableVideoDiffusionPipeline
-import torch
-
-model_id = 'TaiMingLu/GenEx-World-Explorer'
-
-# Load the custom UNet
-unet = UNetSpatioTemporalConditionModel.from_pretrained(
-    model_id,
-    subfolder='unet',
-    torch_dtype=torch.float16,
-    low_cpu_mem_usage=True,
-)
-
-# Load the full pipeline with custom UNet
-pipe = StableVideoDiffusionPipeline.from_pretrained(
-    model_id,
-    unet=unet,
-    low_cpu_mem_usage=True,
-    torch_dtype=torch.float16,
-    local_files_only=True,
-).to('cuda')
-```
-
-## 🔧 Requirements
-
-```
-diffusers>=0.33.1
-transformers
-numpy
-pillow
+---
+base_model:
+- stabilityai/stable-video-diffusion-img2vid-xt-1-1
+pipeline_tag: image-to-video
+---
+
+# GenEx-World-Initializer 🚀🌍
+
+**GenEx World Explorer** is a video generation pipeline built on top of [Stable Video Diffusion (SVD)](https://huggingface.co/stabilityai/stable-video-diffusion-img2vid-xt-1-1).
+. It takes a keyframe, and generates a temporally consistent video. This explorer version builds on SVD with a custom `UNetSpatioTemporalConditionModel`.
+
+The diffuser generate a forward moving path of a panoramic input image, to explore a given scene.
+
+
+![](examples.png)
+
+## 📦 Usage
+
+```python
+from diffusers import UNetSpatioTemporalConditionModel, StableVideoDiffusionPipeline
+import torch
+
+model_id = 'TaiMingLu/GenEx-World-Explorer'
+
+# Load the custom UNet
+unet = UNetSpatioTemporalConditionModel.from_pretrained(
+    model_id,
+    subfolder='unet',
+    torch_dtype=torch.float16,
+    low_cpu_mem_usage=True,
+)
+
+# Load the full pipeline with custom UNet
+pipe = StableVideoDiffusionPipeline.from_pretrained(
+    model_id,
+    unet=unet,
+    low_cpu_mem_usage=True,
+    torch_dtype=torch.float16,
+    local_files_only=True,
+).to('cuda')
+```
+
+## 🔧 Requirements
+
+```
+diffusers>=0.33.1
+transformers
+numpy
+pillow
 ```