File tree 2 files changed +6
-7
lines changed
2 files changed +6
-7
lines changed Original file line number Diff line number Diff line change 1
- import json
2
1
import os
3
2
import tempfile
4
3
7
6
8
7
from faster_whisper import WhisperModel
9
8
9
+ from open_dubbing .utterance import Utterance
10
+
10
11
11
12
class TestCmd :
12
13
@@ -34,11 +35,9 @@ def _update_translation(self, directory):
34
35
file .write (text )
35
36
36
37
def _get_utterances (self , directory ):
37
- metadata_file = os .path .join (directory , "utterance_metadata_cat.json" )
38
- with open (metadata_file , encoding = "utf-8" ) as json_data :
39
- data = json .load (json_data )
40
- utterances = data ["utterances" ]
41
- return utterances
38
+ utterance = Utterance (target_language = "cat" , output_directory = directory )
39
+ utterances , _ , _ = utterance .load_utterances ()
40
+ return utterances
42
41
43
42
def _assert_dubbing_action (self , directory ):
44
43
utterances = self ._get_utterances (directory )
Original file line number Diff line number Diff line change @@ -44,7 +44,7 @@ def _get_file_name(self):
44
44
def load_utterances (self ) -> tuple [Any , PreprocessingArtifacts , Any ]:
45
45
utterance_metadata_file = self ._get_file_name ()
46
46
47
- with open (utterance_metadata_file , "r" ) as file :
47
+ with open (utterance_metadata_file , "r" , encoding = "utf-8" ) as file :
48
48
data = json .load (file )
49
49
utterances = data ["utterances" ]
50
50
preprocessing_output = PreprocessingArtifacts (
You can’t perform that action at this time.
0 commit comments