jhj0517 commited on
Commit
7e7c077
·
1 Parent(s): fb25804

Add test subtitle files

Browse files
Files changed (1) hide show
  1. tests/test_config.py +4 -1
tests/test_config.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  import os
2
 
3
  TEST_FILE_DOWNLOAD_URL = "https://github.com/jhj0517/whisper_flutter_new/raw/main/example/assets/jfk.wav"
@@ -5,5 +7,6 @@ TEST_YOUTUBE_URL = "https://www.youtube.com/watch?v=4WEQtgnBu0I&ab_channel=Andri
5
  TEST_WHISPER_MODEL = "tiny"
6
  TEST_UVR_MODEL = "UVR-MDX-NET-Inst_HQ_4"
7
  TEST_NLLB_MODEL = "facebook/nllb-200-distilled-600M"
8
- TEST_SUBTITLE_PATH = os.path.join("translation_test.srt")
 
9
 
 
1
+ from modules.utils.paths import *
2
+
3
  import os
4
 
5
  TEST_FILE_DOWNLOAD_URL = "https://github.com/jhj0517/whisper_flutter_new/raw/main/example/assets/jfk.wav"
 
7
  TEST_WHISPER_MODEL = "tiny"
8
  TEST_UVR_MODEL = "UVR-MDX-NET-Inst_HQ_4"
9
  TEST_NLLB_MODEL = "facebook/nllb-200-distilled-600M"
10
+ TEST_SUBTITLE_SRT_PATH = os.path.join(WEBUI_DIR, "tests", "test_srt.srt")
11
+ TEST_SUBTITLE_VTT_PATH = os.path.join(WEBUI_DIR, "tests", "test_vtt.vtt")
12