Catch keyboard interrupt in diart.stream (#183)

juanmc2005 · Oct 11, 2023 · 45f8ad9 · 45f8ad9
1 parent 1d1d826
commit 45f8ad9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/diart/console/stream.py b/src/diart/console/stream.py
@@ -122,7 +122,10 @@ def run():
     inference.attach_observers(
         RTTMWriter(audio_source.uri, args.output / f"{audio_source.uri}.rttm")
     )
-    inference()
+    try:
+        inference()
+    except KeyboardInterrupt:
+        pass
 
 
 if __name__ == "__main__":