Spaces:
Sleeping
Sleeping
Merge branch 'oop-refactor' of https://github.com/project-kxkg/project-t into oop-refactor
Browse files- configs/task_config.yaml +7 -3
- src/task.py +1 -1
configs/task_config.yaml
CHANGED
@@ -1,6 +1,10 @@
|
|
1 |
# configuration for each task
|
2 |
-
model:
|
3 |
-
|
4 |
-
output_type:
|
|
|
|
|
|
|
|
|
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.
|
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":
|