ex510 commited on
Commit
bdc1e3e
Β·
verified Β·
1 Parent(s): ccc74e4

Update core/stt.py

Browse files
Files changed (1) hide show
  1. core/stt.py +2 -2
core/stt.py CHANGED
@@ -263,7 +263,7 @@ class SubtitleSegmenter:
263
  # ─────────────────────────────────────────────────────────────────────────────
264
 
265
  class STT:
266
- def __init__(self, model_size="base"):
267
  self.duration = 0
268
  self.model_size = model_size
269
  if not faster_whisper_available:
@@ -370,7 +370,7 @@ class STT:
370
 
371
  segments_iter, info = self.model.transcribe(
372
  video_path,
373
- beam_size = 1,
374
  word_timestamps = True,
375
  language = lang_arg,
376
  task = task_type,
 
263
  # ─────────────────────────────────────────────────────────────────────────────
264
 
265
  class STT:
266
+ def __init__(self, model_size="turbo"):
267
  self.duration = 0
268
  self.model_size = model_size
269
  if not faster_whisper_available:
 
370
 
371
  segments_iter, info = self.model.transcribe(
372
  video_path,
373
+ beam_size = 5,
374
  word_timestamps = True,
375
  language = lang_arg,
376
  task = task_type,