From 46f3eae26ac21f55084996e10d9b994a41baabea Mon Sep 17 00:00:00 2001
From: Eustache Le Bihan <eulebihan@gmail.com>
Date: Mon, 12 Aug 2024 14:55:49 +0000
Subject: [PATCH] listen and play
---
listen_and_play.py | 11 ++++++++++-
1 file changed, 10 insertions(+), 1 deletion(-)
diff --git a/listen_and_play.py b/listen_and_play.py
index 285fe55..cb18fc1 100644
--- a/listen_and_play.py
+++ b/listen_and_play.py
@@ -3,6 +3,8 @@ import numpy as np
import threading
from queue import Queue
from dataclasses import dataclass, field
+from transformers import HfArgumentParser
+
@dataclass
class ListenAndPlayArguments:
@@ -126,4 +128,11 @@ def listen_and_play(
send_socket.close()
recv_socket.close()
- print("Connection closed.")
\ No newline at end of file
+ print("Connection closed.")
+
+
+if __name__ == "__main__":
+ parser = HfArgumentParser((ListenAndPlayArguments,))
+ listen_and_play_kwargs = parser.parse_args_into_dataclasses()
+ listen_and_play(**vars(listen_and_play_kwargs))
+
--
GitLab