JiaenLiu commited on
Commit
6bff4f1
·
1 Parent(s): 2f9051f

in progress

Browse files

Former-commit-id: 1c7b39248f6df226edac1a2d0a73b9caaeedbd86

Files changed (1) hide show
  1. pipeline.py +2 -4
pipeline.py CHANGED
@@ -261,12 +261,10 @@ for sentence, range in tqdm(zip(script_arr, range_arr)):
261
  time.sleep(30)
262
  flag = True
263
  # add read-time output back and modify the post-processing by using one batch as an unit.
264
- print(translate)
265
  srt.set_translation(translate, range, model_name)
266
- add_length = srt.check_len_and_split_range(range)
267
- print(add_length)
268
  srt.realtime_write_srt(f"{RESULT_PATH}/{VIDEO_NAME}/{VIDEO_NAME}_zh.srt",range, add_length,segidx)
269
- srt.realtime_bilingual_write_srt(f"{RESULT_PATH}/{VIDEO_NAME}/{VIDEO_NAME}_bi.srt",range, add_length,segidx)
270
 
271
  # srt.check_len_and_split()
272
  # srt.write_srt_file_translate(f"{RESULT_PATH}/{VIDEO_NAME}/{VIDEO_NAME}_zh.srt")
 
261
  time.sleep(30)
262
  flag = True
263
  # add read-time output back and modify the post-processing by using one batch as an unit.
 
264
  srt.set_translation(translate, range, model_name)
265
+ add_length = srt.check_len_and_split_range(range, threshold)
 
266
  srt.realtime_write_srt(f"{RESULT_PATH}/{VIDEO_NAME}/{VIDEO_NAME}_zh.srt",range, add_length,segidx)
267
+ # srt.realtime_bilingual_write_srt(f"{RESULT_PATH}/{VIDEO_NAME}/{VIDEO_NAME}_bi.srt",range, add_length,segidx)
268
 
269
  # srt.check_len_and_split()
270
  # srt.write_srt_file_translate(f"{RESULT_PATH}/{VIDEO_NAME}/{VIDEO_NAME}_zh.srt")