Update app.py

app.py

@@ -5,6 +5,7 @@ import tempfile
 import os
 import torch
 import gc
+import spaces
 
 # WhisperX配置
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -16,6 +17,7 @@ YT_LENGTH_LIMIT_S = 3600  # 1 hour YouTube files
 # 加载WhisperX模型
 model = whisperx.load_model(MODEL_NAME, device=device, compute_type=compute_type)
 
+@spaces.GPU
 def transcribe(inputs, task):
     if inputs is None:
         raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")