Updated README for inference code

README.md

@@ -5,7 +5,26 @@ tags:
 - pytorch
 library_name: transformers
 ---
-
+## How to Use
+
+```python
+from transformers import AutoModel
+model = AutoModel.from_pretrained(
+    "Nvidia-CMU25/DiffusionVideo2WorldGeneration", 
+    cache_dir="./cache", 
+    trust_remote_code=True, 
+    # force_download=True,
+    # turn on offloading on a low GPU memory machine:
+    # offload_network=True,
+    # offload_tokenizer=True,
+    # offload_text_encoder_model=True,
+    # offload_prompt_upsampler=True,
+    # offload_guardrail_models=True,
+)
+prompt = "Some text prompt to generate a video"
+input_image_or_video_path = "path/to/image.jpeg"
+model(prompt, input_image_or_video_path)
+```
 ![Cosmos Logo](https://github.com/NVIDIA/Cosmos/raw/main/assets/cosmos-logo.png)
 
 --------------------------------------------------------------------------------