Add VideoPrism as library (#1581)

This adds VideoPrism so that download stats are tracked

Relevant links: 
google-deepmind/videoprism#10
https://huggingface.co/google/videoprism

---------

Co-authored-by: Pedro Cuenca <[email protected]>

Author: jenjsun

packages/tasks/src/model-libraries-snippets.ts

@@ -1542,6 +1542,20 @@ image = sana(
 ) `,
 ];
 
+export const videoprism = (model: ModelData): string[] => [
+	`# Install from https://github.com/google-deepmind/videoprism
+import jax
+import jax.numpy as jnp
+from videoprism import models as vp
+
+flax_model = vp.MODELS["${model.id}"]()
+loaded_state = vp.load_pretrained_weights("${model.id}")
+
+@jax.jit
+def forward_fn(inputs, train=False):
+  return flax_model.apply(loaded_state, inputs, train=train)`,
+];
+
 export const vfimamba = (model: ModelData): string[] => [
 	`from Trainer_finetune import Model
 

packages/tasks/src/model-libraries.ts

@@ -1093,6 +1093,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		countDownloads: `path_extension:"pth"`,
 		snippets: snippets.sana,
 	},
+	videoprism: {
+		prettyLabel: "VideoPrism",
+		repoName: "VideoPrism",
+		repoUrl: "https://github.com/google-deepmind/videoprism",
+		countDownloads: `path_extension:"npz"`,
+		snippets: snippets.videoprism,
+	},
 	"vfi-mamba": {
 		prettyLabel: "VFIMamba",
 		repoName: "VFIMamba",