diff --git a/autotranscript/autotranscipt.py b/autotranscript/autotranscipt.py index cbf2c9d..906166a 100644 --- a/autotranscript/autotranscipt.py +++ b/autotranscript/autotranscipt.py @@ -86,7 +86,7 @@ class AutoTranscribe: final_transcript[i] = {"speaker" : diarisation["speakers"][i], "text" : transcript} - return Transcript(transcript, diarisation) + return Transcript(final_transcript) @staticmethod def get_audiofile(audiofile : Union[str, torch.Tensor],