Spaces:
Sleeping
Sleeping
Update core/stt.py
Browse files- 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="
|
| 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 =
|
| 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,
|