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

Update subgen.py

Browse Source 
More housekeeping
This commit is contained in:
McCloudS
2023-10-28 23:15:22 -06:00
committed by GitHub
parent 6850829574
commit a47cd332db
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
subgen/subgen.py
View File

@@ -71,9 +71,9 @@ if transcribe_device == "gpu":
app = FastAPI() app = FastAPI()
model = None model = None
files_to_transcribe = [] files_to_transcribe = []
subextension = f".subgen.{whisper_model}.{namesublang}.srt" subextension = f".subgen.{whisper_model.split('.')[0]}.{namesublang}.srt"
print("Transcriptions are limited to running " + str(concurrent_transcriptions) + " at a time") print(f"Transcriptions are limited to running {str(concurrent_transcriptions)} at a time")
print("Running " + str(whisper_threads) + " threads per transcription") print(f"Running {str(whisper_threads)} threads per transcription")
if debug: if debug:
logging.basicConfig(stream=sys.stderr, level=logging.NOTSET) logging.basicConfig(stream=sys.stderr, level=logging.NOTSET)
@@ -219,7 +219,10 @@ def gen_subtitles(file_path: str, transcribe_or_translate_str: str, front=True)
finally: finally:
if len(files_to_transcribe) == 0: if len(files_to_transcribe) == 0:
logging.debug("Queue is empty, clearing/releasing VRAM") logging.debug("Queue is empty, clearing/releasing VRAM")
del model try:
del model
except Exception as e:
None
gc.collect() gc.collect()
def has_subtitle_language(video_file, target_language): def has_subtitle_language(video_file, target_language):
@@ -345,4 +348,4 @@ if transcribe_folders:
print("Starting webhook!") print("Starting webhook!")
if __name__ == "__main__": if __name__ == "__main__":
import uvicorn import uvicorn
uvicorn.run("subgen:app", host="0.0.0.0", port=int(webhookport), reload=debug) uvicorn.run("subgen:app", host="0.0.0.0", port=int(webhookport), reload=debug, use_colors=True)