initial keras-hub support (#986)

Initial support for Keras-hub:

- generic snippet for loading models
- library definition with metadata

This PR also adds "keras-hub" as one of the "filtered" i.e. top libraries and removes "tf-keras" from the list. This affects the default library filters in "Models" and the library completions in the Model details edition form.

---------

Co-authored-by: Pedro Cuenca <[email protected]>
Co-authored-by: Lucain <[email protected]>
Co-authored-by: Julien Chaumond <[email protected]>
Co-authored-by: vb <[email protected]>

Author: 5 people

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

@@ -364,9 +364,9 @@ model = GLiNER.from_pretrained("${model.id}")`,
 ];
 
 export const keras = (model: ModelData): string[] => [
-	`# Available backend options are: "jax", "tensorflow", "torch".
+	`# Available backend options are: "jax", "torch", "tensorflow".
 import os
-os.environ["KERAS_BACKEND"] = "tensorflow"
+os.environ["KERAS_BACKEND"] = "jax"
 	
 import keras
 
@@ -375,9 +375,9 @@ model = keras.saving.load_model("hf://${model.id}")
 ];
 
 export const keras_nlp = (model: ModelData): string[] => [
-	`# Available backend options are: "jax", "tensorflow", "torch".
+	`# Available backend options are: "jax", "torch", "tensorflow".
 import os
-os.environ["KERAS_BACKEND"] = "tensorflow"
+os.environ["KERAS_BACKEND"] = "jax"
 
 import keras_nlp
 
@@ -386,6 +386,21 @@ backbone = keras_nlp.models.Backbone.from_preset("hf://${model.id}")
 `,
 ];
 
+export const keras_hub = (model: ModelData): string[] => [
+	`# Available backend options are: "jax", "torch", "tensorflow".
+import os
+os.environ["KERAS_BACKEND"] = "jax"
+
+import keras_hub
+
+# Load a task-specific model (*replace CausalLM with your task*)
+model = keras_hub.models.CausalLM.from_preset("hf://${model.id}", dtype="bfloat16")
+
+# Possible tasks are CausalLM, TextToImage, ImageClassifier, ...
+# full list here: https://keras.io/api/keras_hub/models/#api-documentation
+`,
+];
+
 export const llama_cpp_python = (model: ModelData): string[] => [
 	`from llama_cpp import Llama
 

packages/tasks/src/model-libraries.ts

@@ -345,16 +345,23 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		repoUrl: "https://github.com/keras-team/tf-keras",
 		docsUrl: "https://huggingface.co/docs/hub/tf-keras",
 		snippets: snippets.tf_keras,
-		filter: true,
 		countDownloads: `path:"saved_model.pb"`,
 	},
 	"keras-nlp": {
 		prettyLabel: "KerasNLP",
 		repoName: "KerasNLP",
-		repoUrl: "https://keras.io/keras_nlp/",
-		docsUrl: "https://github.com/keras-team/keras-nlp",
+		repoUrl: "https://github.com/keras-team/keras-nlp",
+		docsUrl: "https://keras.io/keras_nlp/",
 		snippets: snippets.keras_nlp,
 	},
+	"keras-hub": {
+		prettyLabel: "KerasHub",
+		repoName: "KerasHub",
+		repoUrl: "https://github.com/keras-team/keras-hub",
+		docsUrl: "https://keras.io/keras_hub/",
+		snippets: snippets.keras_hub,
+		filter: true,
+	},
 	k2: {
 		prettyLabel: "K2",
 		repoName: "k2",