Update app.py

app.py

@@ -3,6 +3,8 @@ import warnings
 import torch
 from transformers import WhisperTokenizer, WhisperForConditionalGeneration, WhisperProcessor
 import soundfile as sf
+from huggingface_hub import spaces
+
 
 warnings.filterwarnings("ignore")
 
@@ -18,7 +20,7 @@ torch_dtype = torch.float32
 # move model to device
 model.to(device)
 
-
+@spaces.GPU
 def transcribe_audio(audio_file):
     audio_input, _ = sf.read(audio_file)
     inputs = processor(audio_input, sampling_rate=16000, return_tensors="pt")
@@ -54,4 +56,5 @@ with iface:
     transcribe_button.click(fn=transcribe_audio, inputs=audio_input, outputs=transcription_output)
 
 # Launch the interface
-iface.launch(share=True, debug=True)
+iface.launch(share=True, debug=True)
+