Quickstart one-line video save + export helper

README.md

@@ -71,6 +71,7 @@ Minimal runtime dependencies: `torch`, `diffusers`, `transformers`, `huggingface
 
 ```python
 from stimulus_synthesis import NevoPipeline
+from stimulus_synthesis.media import save_video, video_to_t_c_h_w
 
 # config (referenced model ids) is fetched from the Hub; weights are pulled on first use
 pipe = NevoPipeline.from_pretrained("epfl-neuroai/NEvo")
@@ -83,8 +84,7 @@ out = pipe(
     seed=0,
 )
 print(out.best_prompt, out.best_score)
-out.best.image   # PIL image
-out.best.video   # list of frames
+save_video(video_to_t_c_h_w(out.best.video), "best_stimulus.mp4")   # save the synthesized video
 ```
 
 ### Targeting a brain region

stimulus_synthesis/media/__init__.py

@@ -1,7 +1,7 @@
 from .asset_decode import DecodedImage, DecodedVideo, decode_image, decode_video
 from .asset_export import AssetExportRecord, export_image, export_video, sha256_file
 from .asset_spec import ImageAssetSpec, VideoAssetSpec
-from .normalize import videos_to_b_t_c_h_w
+from .normalize import video_to_t_c_h_w, videos_to_b_t_c_h_w
 from .video_io import load_video_as_tensor, save_video
 
 __all__ = [
@@ -17,5 +17,6 @@ __all__ = [
     "load_video_as_tensor",
     "save_video",
     "sha256_file",
+    "video_to_t_c_h_w",
     "videos_to_b_t_c_h_w",
 ]