adelorenzo/talk2me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

whisper model to base

Browse Source
This commit is contained in:
Adolfo Delorenzo 2025-04-12 01:10:28 -06:00
parent d46df7939a
commit ae8eaaacc0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app.py
View File

@ -184,7 +184,7 @@ def update_tts_config():
# Load Whisper model # Load Whisper model
logger.info("Loading Whisper model...") logger.info("Loading Whisper model...")
whisper_model = whisper.load_model("medium") whisper_model = whisper.load_model("base")
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
whisper_model = whisper_model.to(device) whisper_model = whisper_model.to(device)
logger.info("Whisper model loaded successfully") logger.info("Whisper model loaded successfully")