Spaces:
Running
Running
jhj0517
commited on
Commit
·
6bc716a
1
Parent(s):
3b69814
Refactor parameter
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -212,19 +212,19 @@ class WhisperBase(ABC):
|
|
212 |
files = get_media_files(input_folder_path)
|
213 |
if isinstance(files, str):
|
214 |
files = [files]
|
215 |
-
if files and
|
216 |
-
files =
|
217 |
|
218 |
files_info = {}
|
219 |
for file in files:
|
220 |
transcribed_segments, time_for_task = self.run(
|
221 |
-
file
|
222 |
progress,
|
223 |
add_timestamp,
|
224 |
*whisper_params,
|
225 |
)
|
226 |
|
227 |
-
file_name, file_ext = os.path.splitext(os.path.basename(file
|
228 |
subtitle, file_path = self.generate_and_write_file(
|
229 |
file_name=file_name,
|
230 |
transcribed_segments=transcribed_segments,
|
|
|
212 |
files = get_media_files(input_folder_path)
|
213 |
if isinstance(files, str):
|
214 |
files = [files]
|
215 |
+
if files and isinstance(files[0], gr.utils.NamedString):
|
216 |
+
files = [file.name for file in files]
|
217 |
|
218 |
files_info = {}
|
219 |
for file in files:
|
220 |
transcribed_segments, time_for_task = self.run(
|
221 |
+
file,
|
222 |
progress,
|
223 |
add_timestamp,
|
224 |
*whisper_params,
|
225 |
)
|
226 |
|
227 |
+
file_name, file_ext = os.path.splitext(os.path.basename(file))
|
228 |
subtitle, file_path = self.generate_and_write_file(
|
229 |
file_name=file_name,
|
230 |
transcribed_segments=transcribed_segments,
|