Update app.py

app.py

@@ -3,9 +3,13 @@ import warnings
 import torch
 from transformers import WhisperTokenizer, WhisperForConditionalGeneration, WhisperProcessor
 import soundfile as sf
+from huggingface_hub import login
 
 warnings.filterwarnings("ignore")
 
+# Ensure you are logged in to Huggingface Hub
+login(token="your_huggingface_token")
+
 # Load tokenizer and model
 tokenizer = WhisperTokenizer.from_pretrained("NbAiLabBeta/nb-whisper-medium")
 model = WhisperForConditionalGeneration.from_pretrained("NbAiLabBeta/nb-whisper-medium")
@@ -15,6 +19,7 @@ processor = WhisperProcessor.from_pretrained("NbAiLabBeta/nb-whisper-medium")
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 torch_dtype = torch.float32
 
+@spaces.GPU
 def transcribe_audio(audio_file):
     audio_input, _ = sf.read(audio_file)
     inputs = processor(audio_input, sampling_rate=16000, return_tensors="pt")
@@ -50,4 +55,4 @@ 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)