diff --git a/core/gallery/parakeet_bundle_entries_test.go b/core/gallery/parakeet_bundle_entries_test.go index 95c96a8ea..c09f69e6b 100644 --- a/core/gallery/parakeet_bundle_entries_test.go +++ b/core/gallery/parakeet_bundle_entries_test.go @@ -25,12 +25,12 @@ var _ = Describe("gallery/index.yaml parakeet-cpp bundle entries", func() { "parakeet-cpp-bundle-small": { "parakeet-cpp/parakeet-bundle-small.gguf", "5f2c6697eb08d4d67e858692d4223a9d911aacc51ae16cf2aee9e760f8a2269a", - []string{"transcript", "vad", "diarization", "sound_classification"}, full, + []string{"transcript", "vad", "diarization", "sound_classification", "speaker_recognition"}, full, }, "parakeet-cpp-bundle-standard": { "parakeet-cpp/parakeet-bundle-standard.gguf", "85304e3bb49de04b84a7d0c76baaa7dedd1a0c02c0920078a99f92c00cffbdf6", - []string{"transcript", "vad", "diarization", "sound_classification"}, full, + []string{"transcript", "vad", "diarization", "sound_classification", "speaker_recognition"}, full, }, "parakeet-cpp-bundle-moondream-redux": { "parakeet-cpp/parakeet-bundle-moondream-redux.gguf", @@ -80,6 +80,16 @@ var _ = Describe("gallery/index.yaml parakeet-cpp bundle entries", func() { } }) + It("declares speaker recognition only where the model has a speaker encoder", func() { + all := entries() + for _, name := range []string{"parakeet-cpp-bundle-small", "parakeet-cpp-bundle-standard", "parakeet-cpp-realtime-scene-speakers"} { + Expect(all[name].Overrides["known_usecases"]).To(ContainElement("speaker_recognition"), name) + } + for _, name := range []string{"parakeet-cpp-bundle-moondream-redux", "parakeet-cpp-tdt_ctc-110m"} { + Expect(all[name].Overrides["known_usecases"]).ToNot(ContainElement("speaker_recognition"), name) + } + }) + It("states that the licence is per component and names each one", func() { all := entries() for name := range want { diff --git a/gallery/index.yaml b/gallery/index.yaml index 7c9d8a766..9e52e70eb 100644 --- a/gallery/index.yaml +++ b/gallery/index.yaml @@ -54528,6 +54528,7 @@ backend: parakeet-cpp known_usecases: - transcript + - speaker_recognition name: parakeet-cpp-realtime-scene-speakers options: - diarization_model:parakeet-cpp/nemotron-3-diarization-q8_0.gguf @@ -54902,6 +54903,7 @@ - vad - diarization - sound_classification + - speaker_recognition name: parakeet-cpp-bundle-small options: - vad:true @@ -54965,6 +54967,7 @@ - vad - diarization - sound_classification + - speaker_recognition name: parakeet-cpp-bundle-standard options: - vad:true