Update README.md

README.md

@@ -18,25 +18,46 @@ It follows by [this paper](https://arxiv.org/abs/2312.10656).
 from diffusers import DiffusionPipeline
 
 # load the pretrained model
-pipeline = DiffusionPipeline.from_pretrained("jadechoghari/VidToMe", trust_remote_code=True, custom_pipeline="jadechoghari/VidToMe", sd_version="depth", device="cuda", float_precision="fp16")
-
-# Edit a video with prompts
-pipeline(
-    video_path="path/to/video.mp4", 
-    video_prompt="A serene beach scene", 
-    edit_prompt="Make the sunset more vibrant", 
-    control_type="depth", 
-    n_timesteps=50
+pipeline = DiffusionPipeline.from_pretrained(
+    "jadechoghari/VidToMe", 
+    trust_remote_code=True, 
+    custom_pipeline="jadechoghari/VidToMe", 
+    sd_version="depth", 
+    device="cuda", 
+    float_precision="fp16"
 )
+
+# specify the input video and output directory
+input_path = "flamingo.mp4"  # Input video file
+work_dir = "output/flamingo"  # Directory for saving results
+
+# set prompts for inversion and generation
+inversion_prompt = "flamingos standing in the water near a tree."
+generation_prompt = {"origami": "rainbow-colored origami flamingos standing in the water near a tree."}
+
+# additional control and parameters
+control_type = "none"  # No extra control, use "depth" if needed
+negative_prompt = ""
+
+# Run the video-to-image editing pipeline
+generated_images = pipeline(
+    video_path="path/to/video.mp4",            # add path to the input video
+    video_prompt=inversion_prompt,    # inversion prompt
+    edit_prompt=generation_prompt,    # edit prompt for generation
+    control_type=control_type         # control type (e.g., "none", "depth")
+)
+
 ```
 
+#### Note: For more control, consider creating a configuration and follow the instructions in the GitHub repository.
+
 ## Applications:
 - Zero-shot video editing for content creators
 - Video transformation using natural language prompts
 - Memory-optimized video generation for longer or complex sequences
 
 **Model Authors:**  
-- Xirui Li  
+- [Xirui Li](https://github.com/lixirui142)  
 - Chao Ma  
 - Xiaokang Yang  
 - Ming-Hsuan Yang