From 2b175b13c4e66662708aae449c50bdff608a4d4d Mon Sep 17 00:00:00 2001
From: erew123 <35898566+erew123@users.noreply.github.com>
Date: Sun, 24 Dec 2023 15:56:44 +0000
Subject: [PATCH] update to find mp3 and flac

Forgot it was only looking for wav's!
---
 finetune.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/finetune.py b/finetune.py
index 3ab2be7..4312c8e 100644
--- a/finetune.py
+++ b/finetune.py
@@ -59,7 +59,7 @@ def create_temporary_file(folder, suffix=".wav"):
     return os.path.join(folder, unique_filename)
 
 def format_audio_list(target_language, whisper_model, out_path, gradio_progress=progress):
-    audio_files = [os.path.join(audio_folder, file) for file in os.listdir(audio_folder) if file.endswith('.wav')]
+    audio_files = [os.path.join(audio_folder, file) for file in os.listdir(audio_folder) if file.endswith(('.mp3', '.flac', '.wav'))]
     buffer=0.2
     eval_percentage=0.15
     speaker_name="coqui"