From b7aeef47731ca81a19c6de616a1f2270a3293a00 Mon Sep 17 00:00:00 2001 From: McCloudS <64094529+McCloudS@users.noreply.github.com> Date: Sat, 10 Feb 2024 19:25:22 -0700 Subject: [PATCH] Add message show if using faster-whisper or HF Transformers --- subgen/subgen.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/subgen/subgen.py b/subgen/subgen.py index 8db9e8c..3b95fe1 100644 --- a/subgen/subgen.py +++ b/subgen/subgen.py @@ -650,6 +650,10 @@ if __name__ == "__main__": print(f"Transcriptions are limited to running {str(concurrent_transcriptions)} at a time") print(f"Running {str(whisper_threads)} threads per transcription") print(f"Using {transcribe_device} to encode") + if hf_transformers: + print(f"Using Hugging Face Transformers") + else: + print(f"Using faster-whisper") if transcribe_folders: transcribe_existing(transcribe_folders) uvicorn.run("subgen:app", host="0.0.0.0", port=int(webhookport), reload=debug, use_colors=True)