Update audio2text.py

vkola-lab · Oct 30, 2024 · bf7276c · bf7276c
1 parent 5466eba
commit bf7276c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/audio2text.py b/scripts/audio2text.py
@@ -30,7 +30,7 @@ def __init__(self, file_dir):
         self.file_dir = file_dir
         self.audio_files = [
             os.path.join(self.file_dir, filename) for filename in os.listdir(self.file_dir)
-            if (filename.endswith(".mp3") or filename.endswith(".m4a") or filename.endswith(".wav"))
+            if filename.endswith((".mp3", ".m4a", ".wav"))
         ]
 
     def __len__(self):