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 |   14 +++++++++++---
 1 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/funasr/utils/version_checker.py b/funasr/utils/version_checker.py
index b89af48..9bc11de 100644
--- a/funasr/utils/version_checker.py
+++ b/funasr/utils/version_checker.py
@@ -1,9 +1,10 @@
-import requests
 from packaging import version
 from funasr import __version__  # Ensure that __version__ is defined in your package's __init__.py
 
 
 def get_pypi_version(package_name):
+    import requests
+
     url = f"https://pypi.org/pypi/{package_name}/json"
     response = requests.get(url)
     if response.status_code == 200:
@@ -13,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