revert

huggingface · Jan 10, 2025 · a6760c6 · a6760c6
1 parent 27a6a73
commit a6760c6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts
@@ -1028,7 +1028,7 @@ wavs = chat.infer(texts, )
 torchaudio.save("output1.wav", torch.from_numpy(wavs[0]), 24000)`,
 ];
 
-export const yolo = (model: ModelData): string[] => [
+export const ultralytics = (model: ModelData): string[] => [
 	`from ultralytics import YOLOv8 # modify the yolo version here
 
 model = YOLOv8.from_pretrained("${model.id}")

diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts
@@ -843,14 +843,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		docsUrl: "https://github.com/jasonppy/VoiceCraft",
 		snippets: snippets.voicecraft,
 	},
-	yolov8: {
-		prettyLabel: "yolo",
-		repoName: "yolo",
+	ultralyrics: {
+		prettyLabel: "ultralyrics",
+		repoName: "ultralyrics",
 		repoUrl: "https://github.com/ultralytics/ultralytics",
 		docsUrl: "https://github.com/ultralytics/ultralytics",
 		filter: false,
 		countDownloads: `path_extension:"pt"`,
-		snippets: snippets.yolo,
+		snippets: snippets.ultralyrics,
 	},
 	yolov10: {
 		prettyLabel: "YOLOv10",