Spaces:
Running
Running
jhj0517
commited on
Commit
·
5f193dc
1
Parent(s):
f12a40c
Refactor return type
Browse files
modules/whisper/base_transcription_pipeline.py
CHANGED
|
@@ -179,7 +179,7 @@ class BaseTranscriptionPipeline(ABC):
|
|
| 179 |
add_timestamp: bool = True,
|
| 180 |
progress=gr.Progress(),
|
| 181 |
*pipeline_params,
|
| 182 |
-
) ->
|
| 183 |
"""
|
| 184 |
Write subtitle file from Files
|
| 185 |
|
|
@@ -250,7 +250,7 @@ class BaseTranscriptionPipeline(ABC):
|
|
| 250 |
result_str = f"Done in {self.format_time(total_time)}! Subtitle is in the outputs folder.\n\n{total_result}"
|
| 251 |
result_file_path = [info['path'] for info in files_info.values()]
|
| 252 |
|
| 253 |
-
return
|
| 254 |
|
| 255 |
except Exception as e:
|
| 256 |
print(f"Error transcribing file: {e}")
|
|
@@ -264,7 +264,7 @@ class BaseTranscriptionPipeline(ABC):
|
|
| 264 |
add_timestamp: bool = True,
|
| 265 |
progress=gr.Progress(),
|
| 266 |
*pipeline_params,
|
| 267 |
-
) ->
|
| 268 |
"""
|
| 269 |
Write subtitle file from microphone
|
| 270 |
|
|
@@ -314,7 +314,7 @@ class BaseTranscriptionPipeline(ABC):
|
|
| 314 |
)
|
| 315 |
|
| 316 |
result_str = f"Done in {self.format_time(time_for_task)}! Subtitle file is in the outputs folder.\n\n{subtitle}"
|
| 317 |
-
return
|
| 318 |
except Exception as e:
|
| 319 |
print(f"Error transcribing mic: {e}")
|
| 320 |
raise
|
|
@@ -327,7 +327,7 @@ class BaseTranscriptionPipeline(ABC):
|
|
| 327 |
add_timestamp: bool = True,
|
| 328 |
progress=gr.Progress(),
|
| 329 |
*pipeline_params,
|
| 330 |
-
) ->
|
| 331 |
"""
|
| 332 |
Write subtitle file from Youtube
|
| 333 |
|
|
@@ -385,7 +385,7 @@ class BaseTranscriptionPipeline(ABC):
|
|
| 385 |
if os.path.exists(audio):
|
| 386 |
os.remove(audio)
|
| 387 |
|
| 388 |
-
return
|
| 389 |
|
| 390 |
except Exception as e:
|
| 391 |
print(f"Error transcribing youtube: {e}")
|
|
|
|
| 179 |
add_timestamp: bool = True,
|
| 180 |
progress=gr.Progress(),
|
| 181 |
*pipeline_params,
|
| 182 |
+
) -> Tuple[str, str]:
|
| 183 |
"""
|
| 184 |
Write subtitle file from Files
|
| 185 |
|
|
|
|
| 250 |
result_str = f"Done in {self.format_time(total_time)}! Subtitle is in the outputs folder.\n\n{total_result}"
|
| 251 |
result_file_path = [info['path'] for info in files_info.values()]
|
| 252 |
|
| 253 |
+
return result_str, result_file_path
|
| 254 |
|
| 255 |
except Exception as e:
|
| 256 |
print(f"Error transcribing file: {e}")
|
|
|
|
| 264 |
add_timestamp: bool = True,
|
| 265 |
progress=gr.Progress(),
|
| 266 |
*pipeline_params,
|
| 267 |
+
) -> Tuple[str, str]:
|
| 268 |
"""
|
| 269 |
Write subtitle file from microphone
|
| 270 |
|
|
|
|
| 314 |
)
|
| 315 |
|
| 316 |
result_str = f"Done in {self.format_time(time_for_task)}! Subtitle file is in the outputs folder.\n\n{subtitle}"
|
| 317 |
+
return result_str, file_path
|
| 318 |
except Exception as e:
|
| 319 |
print(f"Error transcribing mic: {e}")
|
| 320 |
raise
|
|
|
|
| 327 |
add_timestamp: bool = True,
|
| 328 |
progress=gr.Progress(),
|
| 329 |
*pipeline_params,
|
| 330 |
+
) -> Tuple[str, str]:
|
| 331 |
"""
|
| 332 |
Write subtitle file from Youtube
|
| 333 |
|
|
|
|
| 385 |
if os.path.exists(audio):
|
| 386 |
os.remove(audio)
|
| 387 |
|
| 388 |
+
return result_str, file_path
|
| 389 |
|
| 390 |
except Exception as e:
|
| 391 |
print(f"Error transcribing youtube: {e}")
|