Spaces:
Running
Running
jhj0517
commited on
Commit
·
0facd17
1
Parent(s):
2a0e0ea
refactoring
Browse files
modules/faster_whisper_inference.py
CHANGED
@@ -252,14 +252,14 @@ class FasterWhisperInference(BaseInterface):
|
|
252 |
self.update_model(params.model_size, params.compute_type, progress)
|
253 |
|
254 |
if params.lang == "Automatic Detection":
|
255 |
-
lang = None
|
256 |
else:
|
257 |
language_code_dict = {value: key for key, value in whisper.tokenizer.LANGUAGES.items()}
|
258 |
-
lang = language_code_dict[params.lang]
|
259 |
|
260 |
segments, info = self.model.transcribe(
|
261 |
audio=audio,
|
262 |
-
language=lang,
|
263 |
task="translate" if params.is_translate and self.current_model_size in self.translatable_models else "transcribe",
|
264 |
beam_size=params.beam_size,
|
265 |
log_prob_threshold=params.log_prob_threshold,
|
|
|
252 |
self.update_model(params.model_size, params.compute_type, progress)
|
253 |
|
254 |
if params.lang == "Automatic Detection":
|
255 |
+
params.lang = None
|
256 |
else:
|
257 |
language_code_dict = {value: key for key, value in whisper.tokenizer.LANGUAGES.items()}
|
258 |
+
params.lang = language_code_dict[params.lang]
|
259 |
|
260 |
segments, info = self.model.transcribe(
|
261 |
audio=audio,
|
262 |
+
language=params.lang,
|
263 |
task="translate" if params.is_translate and self.current_model_size in self.translatable_models else "transcribe",
|
264 |
beam_size=params.beam_size,
|
265 |
log_prob_threshold=params.log_prob_threshold,
|