Spaces:
Running
Running
jhj0517
commited on
Commit
•
e65592d
1
Parent(s):
30d7c58
refactor
Browse files
modules/faster_whisper_inference.py
CHANGED
@@ -82,9 +82,6 @@ class FasterWhisperInference(BaseInterface):
|
|
82 |
try:
|
83 |
self.update_model_if_needed(model_size=model_size, compute_type=compute_type, progress=progress)
|
84 |
|
85 |
-
if lang == "Automatic Detection":
|
86 |
-
lang = None
|
87 |
-
|
88 |
files_info = {}
|
89 |
for fileobj in fileobjs:
|
90 |
transcribed_segments, time_for_task = self.transcribe(
|
@@ -176,9 +173,6 @@ class FasterWhisperInference(BaseInterface):
|
|
176 |
try:
|
177 |
self.update_model_if_needed(model_size=model_size, compute_type=compute_type, progress=progress)
|
178 |
|
179 |
-
if lang == "Automatic Detection":
|
180 |
-
lang = None
|
181 |
-
|
182 |
progress(0, desc="Loading Audio from Youtube..")
|
183 |
yt = get_ytdata(youtubelink)
|
184 |
audio = get_ytaudio(yt)
|
@@ -268,9 +262,6 @@ class FasterWhisperInference(BaseInterface):
|
|
268 |
try:
|
269 |
self.update_model_if_needed(model_size=model_size, compute_type=compute_type, progress=progress)
|
270 |
|
271 |
-
if lang == "Automatic Detection":
|
272 |
-
lang = None
|
273 |
-
|
274 |
progress(0, desc="Loading Audio..")
|
275 |
|
276 |
transcribed_segments, time_for_task = self.transcribe(
|
@@ -338,7 +329,10 @@ class FasterWhisperInference(BaseInterface):
|
|
338 |
elapsed time for transcription
|
339 |
"""
|
340 |
start_time = time.time()
|
341 |
-
|
|
|
|
|
|
|
342 |
language_code_dict = {value: key for key, value in whisper.tokenizer.LANGUAGES.items()}
|
343 |
lang = language_code_dict[lang]
|
344 |
segments, info = self.model.transcribe(
|
|
|
82 |
try:
|
83 |
self.update_model_if_needed(model_size=model_size, compute_type=compute_type, progress=progress)
|
84 |
|
|
|
|
|
|
|
85 |
files_info = {}
|
86 |
for fileobj in fileobjs:
|
87 |
transcribed_segments, time_for_task = self.transcribe(
|
|
|
173 |
try:
|
174 |
self.update_model_if_needed(model_size=model_size, compute_type=compute_type, progress=progress)
|
175 |
|
|
|
|
|
|
|
176 |
progress(0, desc="Loading Audio from Youtube..")
|
177 |
yt = get_ytdata(youtubelink)
|
178 |
audio = get_ytaudio(yt)
|
|
|
262 |
try:
|
263 |
self.update_model_if_needed(model_size=model_size, compute_type=compute_type, progress=progress)
|
264 |
|
|
|
|
|
|
|
265 |
progress(0, desc="Loading Audio..")
|
266 |
|
267 |
transcribed_segments, time_for_task = self.transcribe(
|
|
|
329 |
elapsed time for transcription
|
330 |
"""
|
331 |
start_time = time.time()
|
332 |
+
|
333 |
+
if lang == "Automatic Detection":
|
334 |
+
lang = None
|
335 |
+
else:
|
336 |
language_code_dict = {value: key for key, value in whisper.tokenizer.LANGUAGES.items()}
|
337 |
lang = language_code_dict[lang]
|
338 |
segments, info = self.model.transcribe(
|