Update app.py

app.py

@@ -1,8 +1,14 @@
-import gradio as gr
+import torch
 import whisperx
 
-# 加载WhisperX模型
-model = whisperx.load_model("large-v3", device="cpu")  # 如果有GPU，可将"cpu"改为"cuda"
+# 检测设备类型
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# 设置计算类型
+compute_type = "float16" if device == "cuda" else "int8"
+
+# 加载 WhisperX 模型
+model = whisperx.load_model("large-v3", device=device, compute_type=compute_type)
 
 def transcribe(audio_path):
     # 使用WhisperX进行转录