coosld/MoneyPrinterTurbo
1
0
Fork 0
mirror of https://github.com/harry0703/MoneyPrinterTurbo.git synced 2026-02-21 08:27:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #320 from harry0703/dev

Browse Source 
1. Added exception handling when loading the Whisper model. 2. Check if the file exists before verifying the subtitles.
This commit is contained in:
Harry 2024-04-26 11:15:17 +08:00 committed by GitHub
commit b08b79f9cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
app/services/subtitle.py
View File

@ -24,9 +24,18 @@ def create(audio_file, subtitle_file: str = ""):
model_path = model_size
logger.info(f"loading model: {model_path}, device: {device}, compute_type: {compute_type}")
model = WhisperModel(model_size_or_path=model_path,
device=device,
compute_type=compute_type)
try:
model = WhisperModel(model_size_or_path=model_path,
device=device,
compute_type=compute_type)
except Exception as e:
logger.error(f"failed to load model: {e} \n\n"
f"********************************************\n"
f"this may be caused by network issue. \n"
f"please download the model manually and put it in the 'models' folder. \n"
f"see [README.md FAQ](https://github.com/harry0703/MoneyPrinterTurbo) for more details.\n"
f"********************************************\n\n")
return None
logger.info(f"start, output file: {subtitle_file}")
if not subtitle_file:
@ -120,6 +129,9 @@ def create(audio_file, subtitle_file: str = ""):
def file_to_subtitles(filename):
if not filename or not os.path.isfile(filename):
return []
times_texts = []
current_times = None
current_text = ""