Spaces:
Running
Running
File size: 649 Bytes
7e7c077 1e38b6e 88d2794 1e38b6e 3b69814 a17a924 1e38b6e 2952821 7e7c077 1e38b6e 88d2794 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from modules.utils.paths import *
import os
import torch
TEST_FILE_DOWNLOAD_URL = "https://github.com/jhj0517/whisper_flutter_new/raw/main/example/assets/jfk.wav"
TEST_FILE_PATH = os.path.join(WEBUI_DIR, "tests", "jfk.wav")
TEST_YOUTUBE_URL = "https://www.youtube.com/watch?v=4WEQtgnBu0I&ab_channel=AndriaFitzer"
TEST_WHISPER_MODEL = "tiny"
TEST_UVR_MODEL = "UVR-MDX-NET-Inst_HQ_4"
TEST_NLLB_MODEL = "facebook/nllb-200-distilled-600M"
TEST_SUBTITLE_SRT_PATH = os.path.join(WEBUI_DIR, "tests", "test_srt.srt")
TEST_SUBTITLE_VTT_PATH = os.path.join(WEBUI_DIR, "tests", "test_vtt.vtt")
def is_cuda_available():
return torch.cuda.is_available()
|