diff --git a/autotranscript/diarisation.py b/autotranscript/diarisation.py
index 3b64fac..ff3ead0 100644
--- a/autotranscript/diarisation.py
+++ b/autotranscript/diarisation.py
@@ -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