From de1ca223976e4993dd2f2fcd5276a5bf3c556f57 Mon Sep 17 00:00:00 2001
From: Jaikinator <schmieder.jacob@web.de>
Date: Fri, 16 Jun 2023 12:09:53 +0200
Subject: [PATCH] added dict as input type

---
 autotranscript/diarisation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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