diff --git a/conversion_simple_fr.py b/conversion_simple_fr.py index a7c894b..73e8f3c 100644 --- a/conversion_simple_fr.py +++ b/conversion_simple_fr.py @@ -7,8 +7,8 @@ import wave SetLogLevel(0) -if not os.path.exists("models/fr"): - print ("Please download the model from https://alphacephei.com/vosk/models and unpack as 'models' in the current folder.") +if not os.path.exists("model"): + print ("Please download the model from https://alphacephei.com/vosk/models and unpack as 'model' in the current folder.") exit (1) wf = wave.open(sys.argv[1], "rb") @@ -16,7 +16,7 @@ if wf.getnchannels() != 1 or wf.getsampwidth() != 2 or wf.getcomptype() != "NONE print ("Audio file must be WAV format mono PCM.") exit (1) -model = Model("models/fr") +model = Model("model") rec = KaldiRecognizer(model, wf.getframerate()) while True: diff --git a/install.sh b/install.sh index 7536d8c..0e60826 100644 --- a/install.sh +++ b/install.sh @@ -25,17 +25,15 @@ pip3 --version echo -e " ${reset}" pip3 -v install vosk echo "${green}########### récupération du modèle de reconnaissance en Français sous licence aGPL, taille: 1.6Go. Choix des modèles disponibles: https://alphacephei.com/vosk/models ${reset}" -echo " " - mkdir -p models/fr echo -e "${green}########### Procéder au téléchargement du modèle Français (1.6go) pour transcrire les textes ?${reset} (écrivez o pour oui et faites entrée pour valider) ${reset}" read proceed if [[ $proceed == o* ]]; then echo "C'est parti." wget https://alphacephei.com/vosk/models/vosk-model-fr-0.6-linto-2.2.0.zip echo -e "${green}########### téléchargement du modèle OK ${reset}" - unzip vosk-model-fr-0.6-linto-2.2.0.zip -d models/fr + unzip vosk-model-fr-0.6-linto-2.2.0.zip + ln -s vosk-model-fr-0.6-linto-2.2.0 model echo -e "${green}########### décompression du modèle OK ${reset}" - ls -l models/fr else echo -e "${green}########### vous n'avez pas souhaité télécharger le modèle ${reset}" echo -e "${green}########### fin de l'installation sans télécharger de modèle de langue ${reset}"