Fixed @Jojobarthold2 hallo!!!

1 files changed, 2 insertions, 1 deletions

diff --git a/swr2_asr/train.py b/swr2_asr/train.py
index f61776a..df9ffa2 100644
--- a/swr2_asr/train.py
+++ b/swr2_asr/train.py
@@ -1,11 +1,12 @@
 """Training script for the ASR model."""
 from AudioLoader.speech.mls import MultilingualLibriSpeech
+import os
 
 
 def main() -> None:
     """Main function."""
     dataset = MultilingualLibriSpeech(
-        "data", "mls_polish_opus", split="train", download=True
+        "data", "mls_polish_opus", split="train", download=(not os.path.isdir("data"))
     )
 
     print(dataset[1])