From 23e7ddebccd3b05cf7ef89809bcfe565ad6dfa1f Mon Sep 17 00:00:00 2001
From: majic31 <majic31@163.com>
Date: 星期二, 24 十二月 2024 10:00:14 +0800
Subject: [PATCH] Fix the variable name (#2328)

---
 funasr/utils/version_checker.py |   11 +++++++++--
 1 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/funasr/utils/version_checker.py b/funasr/utils/version_checker.py
index 7597530..9bc11de 100644
--- a/funasr/utils/version_checker.py
+++ b/funasr/utils/version_checker.py
@@ -14,12 +14,19 @@
         raise Exception("Failed to retrieve version information from PyPI.")
 
 
-def check_for_update():
+def check_for_update(disable=False):
     current_version = version.parse(__version__)
+    print(f"funasr version: {current_version}.")
+    if disable:
+        return
+
+    print(
+        "Check update of funasr, and it would cost few times. You may disable it by set `disable_update=True` in AutoModel"
+    )
     pypi_version = get_pypi_version("funasr")
 
     if current_version < pypi_version:
-        print(f"New version available: {pypi_version}. Your current version is {current_version}.")
+        print(f"New version is available: {pypi_version}.")
         print('Please use the command "pip install -U funasr" to upgrade.')
     else:
         print(f"You are using the latest version of funasr-{current_version}")

--
Gitblit v1.9.1