update 20250403

api_chat/asr.py

@@ -6,6 +6,9 @@ from dotenv import load_dotenv
 from kafka import KafkaConsumer
 import asyncio
 
+# 在导入其他库之前设置
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
+
 # 设置要使用的GPU ID
 GPU_ID = 1  # 修改这个值来选择要使用的GPU
 
@@ -16,7 +19,7 @@ os.environ["CUDA_VISIBLE_DEVICES"] = str(GPU_ID)
 load_dotenv()
 
 print("正在加载Whisper模型...")
-model = whisper.load_model("large-v3")
+model = whisper.load_model("small")
 print("Whisper模型加载完成。")
 
 # Kafka配置