Utiliser SpeechBrain sur Hugging Face

speechbrain est un kit d’outils conversationnel open source et tout-en-un pour l’audio et la parole. L’objectif est de créer un kit d’outils unique, flexible et convivial, permettant de développer facilement des technologies de pointe en matière de parole, notamment des systèmes de reconnaissance vocale, de reconnaissance de locuteur, d’amélioration de la parole, de séparation de la parole, d’identification de langue, de traitement du signal multi-microphones, et bien d’autres.

Explorer SpeechBrain dans le Hub

Vous pouvez trouver les modèles speechbrain en filtrant à gauche de la page des modèles.

Tous les modèles du Hub sont dotés des fonctionnalités suivantes :

1. Une carte de modèle générée automatiquement avec une brève description.
2. Balises de métadonnées qui facilitent la découvrabilité grâce à des informations telles que la langue, la licence, le papier, etc.
3. Un widget interactif que vous pouvez utiliser pour tester le modèle directement dans le navigateur.
4. Un widget Inference Providers permettant d’effectuer des requêtes d’inférence.

Utilisation de modèles existants

speechbrain propose différentes interfaces pour gérer des modèles préentraînés pour différentes tâches, telles que EncoderClassifier, EncoderClassifier, SepformerSeparation et SpectralMaskEnhancement. Ces classes disposent d’une méthode from_hparams que vous pouvez utiliser pour charger un modèle depuis le Hub

Voici un exemple pour exécuter une inférence de reconnaissance de sons dans des environnements urbains.

import torchaudio
from speechbrain.pretrained import EncoderClassifier

classifier = EncoderClassifier.from_hparams(
    source="speechbrain/urbansound8k_ecapa"
)
out_prob, score, index, text_lab = classifier.classify_file('speechbrain/urbansound8k_ecapa/dog_bark.wav')

Si vous souhaitez voir comment charger un modèle spécifique, vous pouvez cliquer sur Use in speechbrain et un extrait de code fonctionnel vous sera fourni pour le charger !

Ressources supplémentaires

• Site web de SpeechBrain website.
• Documentation de SpeechBrain.