add gpu

app.py

@@ -15,7 +15,7 @@ def init_model(model_dir_aed):
     if asr_model_aed is None:
         asr_model_aed = FireRedAsr.from_pretrained("aed", model_dir_aed)
 
-
+@spaces.GPU(duration=20)
 def asr_inference(audio_file):
     if not audio_file:
         return "Please upload a wav file"
@@ -25,7 +25,7 @@ def asr_inference(audio_file):
         batch_uttid,
         batch_wav_path,
         {
-            "use_gpu": False,
+            "use_gpu": True,
             "beam_size": 3,
             "nbest": 1,
             "decode_max_len": 0,