From 0f3d2d1266fec1b1666be1a4e0a2d234eb152567 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 11 十二月 2024 14:21:57 +0800
Subject: [PATCH] v1.1.17

---
 funasr/utils/version_checker.py |   16 ++++++++++++----
 1 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/funasr/utils/version_checker.py b/funasr/utils/version_checker.py
index 1a6970e..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("You are using the latest version of funasr.")
+        print(f"You are using the latest version of funasr-{current_version}")

--
Gitblit v1.9.1