fix the vector model type error, test=doc

zh794390558 · May 25, 2022 · be8a78a · be8a78a
1 parent a560597
commit be8a78a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/demos/audio_content_search/conf/acs_application.yaml b/demos/audio_content_search/conf/acs_application.yaml
@@ -28,6 +28,7 @@ acs_python:
     word_list: "./conf/words.txt"
     sample_rate: 16000
     device: 'cpu' # set 'gpu:id' or 'cpu'
+    ping_timeout: 100 # seconds
 
 
 

diff --git a/docs/source/released_model.md b/docs/source/released_model.md
@@ -82,7 +82,7 @@ PANN | ESC-50 |[pann-esc50](../../examples/esc50/cls0)|[esc50_cnn6.tar.gz](https
 
 Model Type | Dataset| Example Link | Pretrained Models | Static Models 
 :-------------:| :------------:| :-----: | :-----: | :-----:
-PANN | VoxCeleb| [voxceleb_ecapatdnn](https://github.com/PaddlePaddle/PaddleSpeech/tree/develop/examples/voxceleb/sv0) | [ecapatdnn.tar.gz](https://paddlespeech.bj.bcebos.com/vector/voxceleb/sv0_ecapa_tdnn_voxceleb12_ckpt_0_2_1.tar.gz) | -
+ECAPA-TDNN | VoxCeleb| [voxceleb_ecapatdnn](https://github.com/PaddlePaddle/PaddleSpeech/tree/develop/examples/voxceleb/sv0) | [ecapatdnn.tar.gz](https://paddlespeech.bj.bcebos.com/vector/voxceleb/sv0_ecapa_tdnn_voxceleb12_ckpt_0_2_1.tar.gz) | -
 
 ## Punctuation Restoration Models
 Model Type | Dataset| Example Link | Pretrained Models

diff --git a/paddlespeech/server/engine/acs/python/acs_engine.py b/paddlespeech/server/engine/acs/python/acs_engine.py
@@ -46,6 +46,9 @@ def init(self, config: dict):
         try:
             self.config = config
             self.device = self.config.get("device", paddle.get_device())
+
+            # websocket default ping timeout is 20 seconds
+            self.ping_timeout = self.config.get("ping_timeout", 20)
             paddle.set_device(self.device)
             logger.info(f"ACS Engine set the device: {self.device}")
 
@@ -97,8 +100,8 @@ def get_asr_content(self, audio_data):
             logger.error("No asr server, please input valid ip and port")
             return ""
         ws = websocket.WebSocket()
-        ws.connect(self.url)
-        # with websocket.WebSocket.connect(self.url) as ws:
+        logger.info(f"set the ping timeout: {self.ping_timeout} seconds")
+        ws.connect(self.url, ping_timeout=self.ping_timeout)
         audio_info = json.dumps(
             {
                 "name": "test.wav",