Eason Lu commited on
Commit
d5fcd7b
·
2 Parent(s): 317ffc4 446f373

Merge branch 'oop-refactor' of https://github.com/project-kxkg/project-t into oop-refactor

Browse files
Files changed (2) hide show
  1. configs/task_config.yaml +7 -3
  2. src/task.py +1 -1
configs/task_config.yaml CHANGED
@@ -1,6 +1,10 @@
1
  # configuration for each task
2
- model: "gpt-4"
3
- local_dump: ./local_dump
4
- output_type: srt
 
 
 
 
5
  target_lang: CN
6
  field: SC2
 
1
  # configuration for each task
2
+ model: gpt-4
3
+ # output type that user receive
4
+ output_type:
5
+ subtitle: srt
6
+ video: False
7
+ bilingal: False
8
+ source_lang: EN
9
  target_lang: CN
10
  field: SC2
src/task.py CHANGED
@@ -111,7 +111,7 @@ class Task:
111
  self.SRT_Script.form_whole_sentence()
112
  # self.SRT_Script.spell_check_term()
113
  self.SRT_Script.correct_with_force_term()
114
- processed_srt_path_en = str(Path(self.srt_path).with_suffix('')) + '_processed.srt'
115
  self.SRT_Script.write_srt_file_src(processed_srt_path_en)
116
 
117
  if self.output_type == "ass":
 
111
  self.SRT_Script.form_whole_sentence()
112
  # self.SRT_Script.spell_check_term()
113
  self.SRT_Script.correct_with_force_term()
114
+ processed_srt_path_en = str(Path(self.task_local_dir).with_suffix('')) + '_processed.srt'
115
  self.SRT_Script.write_srt_file_src(processed_srt_path_en)
116
 
117
  if self.output_type == "ass":