added dict as input type

2023-06-16 12:09:53 +02:00
parent 29e8a229dc
commit de1ca22397
1 changed files with 1 additions and 1 deletions
@@ -15,7 +15,7 @@ class Diariser:

        self.model = model

-    def diarization(self, audiofile : Union[str, Tensor] ,
+    def diarization(self, audiofile : Union[str, Tensor, dict] ,
                    *args, **kwargs) -> Annotation:
        """
        Diarization of audio file