diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index 1a0f43372..a48fb71c0 100644 --- 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 index 6e09a5585..5a274c5eb 100644 --- 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",