dasemu/Transcriptarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update subgen.py

Browse Source
This commit is contained in:
McCloudS
2024-03-19 11:49:52 -06:00
committed by GitHub
parent 0ce705f0ca
commit 1af00d962f
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
subgen/subgen.py
View File

@@ -23,8 +23,6 @@ import av
import ffmpeg import ffmpeg
import whisper import whisper
import re import re
from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler
def convert_to_bool(in_bool): def convert_to_bool(in_bool):
if isinstance(in_bool, bool): if isinstance(in_bool, bool):
@@ -53,6 +51,7 @@ use_path_mapping = convert_to_bool(os.getenv('USE_PATH_MAPPING', False))
path_mapping_from = os.getenv('PATH_MAPPING_FROM', r'/tv') path_mapping_from = os.getenv('PATH_MAPPING_FROM', r'/tv')
path_mapping_to = os.getenv('PATH_MAPPING_TO', r'/Volumes/TV') path_mapping_to = os.getenv('PATH_MAPPING_TO', r'/Volumes/TV')
model_location = os.getenv('MODEL_PATH', './models') model_location = os.getenv('MODEL_PATH', './models')
monitor = convert_to_bool(os.getenv('MONITOR', False))
transcribe_folders = os.getenv('TRANSCRIBE_FOLDERS', '') transcribe_folders = os.getenv('TRANSCRIBE_FOLDERS', '')
transcribe_or_translate = os.getenv('TRANSCRIBE_OR_TRANSLATE', 'transcribe') transcribe_or_translate = os.getenv('TRANSCRIBE_OR_TRANSLATE', 'transcribe')
force_detected_language_to = os.getenv('FORCE_DETECTED_LANGUAGE_TO', '') force_detected_language_to = os.getenv('FORCE_DETECTED_LANGUAGE_TO', '')
@@ -65,6 +64,10 @@ append = convert_to_bool(os.getenv('APPEND', False))
if transcribe_device == "gpu": if transcribe_device == "gpu":
transcribe_device = "cuda" transcribe_device = "cuda"
if monitor:
from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler
app = FastAPI() app = FastAPI()
model = None model = None
files_to_transcribe = [] files_to_transcribe = []