Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -422,7 +422,8 @@ class WhisperBase(ABC):
|
|
422 |
temp_subtitle = info["subtitle"]
|
423 |
temp_subtitle.replace("\t","□")
|
424 |
temp_subtitle.replace("\n","■")
|
425 |
-
total_result += title_line
|
|
|
426 |
|
427 |
total_time += info["time_for_task"]
|
428 |
|
@@ -431,7 +432,9 @@ class WhisperBase(ABC):
|
|
431 |
total_info = f"Processed {file_count} {'file' if file_count == 1 else 'files'} in {self.format_time((time_end-time_start).total_seconds())}"
|
432 |
|
433 |
result_str = total_result.rstrip("\n")
|
|
|
434 |
result_str = self.transform_text_to_list(result_str,"□","■")
|
|
|
435 |
result_file_path = [info['path'] for info in files_to_download.values()]
|
436 |
|
437 |
return [result_str,result_file_path,total_info]
|
|
|
422 |
temp_subtitle = info["subtitle"]
|
423 |
temp_subtitle.replace("\t","□")
|
424 |
temp_subtitle.replace("\n","■")
|
425 |
+
total_result += title_line
|
426 |
+
total result += temp_subtitle
|
427 |
|
428 |
total_time += info["time_for_task"]
|
429 |
|
|
|
432 |
total_info = f"Processed {file_count} {'file' if file_count == 1 else 'files'} in {self.format_time((time_end-time_start).total_seconds())}"
|
433 |
|
434 |
result_str = total_result.rstrip("\n")
|
435 |
+
print("Pre: "+result_str)
|
436 |
result_str = self.transform_text_to_list(result_str,"□","■")
|
437 |
+
print("Post: "+result_str)
|
438 |
result_file_path = [info['path'] for info in files_to_download.values()]
|
439 |
|
440 |
return [result_str,result_file_path,total_info]
|