Update subgen.py

This commit is contained in:
McCloudS
2024-02-05 11:26:15 -07:00
committed by GitHub
parent fe004d31b5
commit ff8b465ebd

View File

@@ -53,6 +53,7 @@ transcribe_or_translate = os.getenv('TRANSCRIBE_OR_TRANSLATE', 'translate')
compute_type = os.getenv('COMPUTE_TYPE', 'auto') compute_type = os.getenv('COMPUTE_TYPE', 'auto')
if transcribe_device == "gpu": if transcribe_device == "gpu":
transcribe_device = "cuda" transcribe_device = "cuda"
transcribe_folders = transcribe_folders.split("|")
app = FastAPI() app = FastAPI()
model = None model = None
@@ -485,7 +486,6 @@ def transcribe_existing():
print("Finished searching and queueing files for transcription") print("Finished searching and queueing files for transcription")
if transcribe_folders: if transcribe_folders:
transcribe_folders = transcribe_folders.split("|")
transcribe_existing() transcribe_existing()
whisper_languages = { whisper_languages = {