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,
|