Update handler.py

handler.py

@@ -12,6 +12,9 @@ from transformers import WhisperForConditionalGeneration, pipeline
 def install(package):
     subprocess.check_call([sys.executable, "-m", "pip", "install", package])
 
+# 更新 pip 并安装指定版本的库
+subprocess.run(["pip", "install", "--upgrade", "pip"], check=True)
+subprocess.run(["pip", "install", "--upgrade", "huggingface_hub"], check=True)
 # try:
 #     import bitsandbytes
 # except ImportError:
@@ -51,9 +54,6 @@ class EndpointHandler:
 
 # 确保在安装依赖项之后再实例化和加载模型
 if __name__ == "__main__":
-    # 更新 pip 并安装指定版本的库
-    subprocess.run(["pip", "install", "--upgrade", "pip"], check=True)
-    subprocess.run(["pip", "install", "--upgrade", "huggingface_hub"], check=True)
     try:
         # Install accelerate
         pip.main(['install', 'accelerate'])