Update app.py

app.py

@@ -6,6 +6,7 @@ import torch
 from transformers import WhisperTokenizer, WhisperForConditionalGeneration, WhisperProcessor
 import soundfile as sf
 import ffmpeg
+import spaces
 
 warnings.filterwarnings("ignore")
 
@@ -26,6 +27,8 @@ def convert_audio_format(audio_path):
     ffmpeg.input(audio_path).output(output_path, format='wav', ar='16000').run(overwrite_output=True)
     return output_path
 
+    
+@spaces.GPU 
 def transcribe_audio(audio_file, batch_size=4):
     audio_path = convert_audio_format(audio_file)
     audio_input, sample_rate = sf.read(audio_path)