Update app.py

app.py

@@ -15,6 +15,18 @@ import gradio as gr
 
 from config import model_config
 
+try:
+    import spaces
+    USING_SPACES = True
+except ImportError:
+    USING_SPACES = False
+
+def gpu_decorator(func):
+    if USING_SPACES:
+        return spaces.GPU(func)
+    else:
+        return func
+        
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 model_dir = snapshot_download(model_config['model_dir'])
 
@@ -31,6 +43,7 @@ model = AutoModel(
     device=device,
 )
 
+@gpu_decorator
 def transcribe_audio(file_path, vad_model="fsmn-vad", vad_kwargs='{"max_single_segment_time": 15000}', 
                      batch_size=1, language="auto", use_itn=True, batch_size_s=60, 
                      merge_vad=True, merge_length_s=15, batch_size_threshold_s=50,