From 4f224c88068b66bcb6f81570da59d99c9bba8288 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 26 一月 2024 16:02:14 +0800
Subject: [PATCH] python-websocket funasr1.0 (#1310)

---
 funasr/auto/auto_model.py |   13 +++++++------
 1 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py
index 9d957a3..d072219 100644
--- a/funasr/auto/auto_model.py
+++ b/funasr/auto/auto_model.py
@@ -88,7 +88,8 @@
 class AutoModel:
     
     def __init__(self, **kwargs):
-        tables.print()
+        if kwargs.get("disable_log", False):
+            tables.print()
         
         model, kwargs = self.build_model(**kwargs)
         
@@ -426,10 +427,10 @@
                                  f"time_escape: {time_escape_total_per_sample:0.3f}")
 
 
-        end_total = time.time()
-        time_escape_total_all_samples = end_total - beg_total
-        print(f"rtf_avg_all: {time_escape_total_all_samples / time_speech_total_all_samples:0.3f}, "
-                             f"time_speech_all: {time_speech_total_all_samples: 0.3f}, "
-                             f"time_escape_all: {time_escape_total_all_samples:0.3f}")
+        # end_total = time.time()
+        # time_escape_total_all_samples = end_total - beg_total
+        # print(f"rtf_avg_all: {time_escape_total_all_samples / time_speech_total_all_samples:0.3f}, "
+        #                      f"time_speech_all: {time_speech_total_all_samples: 0.3f}, "
+        #                      f"time_escape_all: {time_escape_total_all_samples:0.3f}")
         return results_ret_list
 

--
Gitblit v1.9.1