whisper

requirements.txt

@@ -1,10 +1,10 @@
 pyodps
 moviepy
 requests
-openai_whisper
 zhconv
 oss2
 pandas
 aliyun_python_sdk
+aliyun-log-python-sdk
 odps
-git+https://github.com/openai/whisper.git
+whisper

whisper_asr.py

@@ -4,7 +4,7 @@ from zhconv import convert
 
 def get_whisper_asr(audio, model='medium'):
     try:
-        model = whisper.load_model(model)
+        model = whisper.load_model(model, device='cuda')
         result = model.transcribe(audio, language='Chinese')
         # 简繁转换
         # zh-cn 大陆简体