Fix JavaScript (huggingface#617)

johko · Mar 16, 2022 · d234f0d · d234f0d
1 parent 8488f77
commit d234f0d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/js/src/lib/interfaces/Libraries.ts b/js/src/lib/interfaces/Libraries.ts
@@ -262,7 +262,7 @@ import fasttext
 
 model = fasttext.load_model(hf_hub_download("${model.id}", "model.bin"))`;
 
-const stable-baselines3 = (model: ModelData) =>
+const stableBaselines3 = (model: ModelData) =>
 	`from huggingface_sb3 import load_from_hub
 	checkpoint = load_from_hub(
     		repo_id="${model.id}",
@@ -380,7 +380,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS: { [key in keyof typeof ModelLibrary]:
 		btnLabel: "stable-baselines3",
 		repoName: "stable-baselines3",
 		repoUrl:  "https://github.com/huggingface/huggingface_sb3",
-		snippet:  stable-baselines3,
+		snippet:  stableBaselines3,
 	},
 } as const;