Update app.py

app.py

@@ -59,7 +59,8 @@ if not torch.cuda.is_available():
     raise RuntimeError("CUDA not available. Go look for a GPU.")
 
 # Set GPU
-device = "cuda"  
+dtype = torch.bfloat32
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 def convert_to_wav(filepath):
     _, file_ending = os.path.splitext(f'{filepath}')
@@ -67,7 +68,7 @@ def convert_to_wav(filepath):
     os.system(f'ffmpeg -i "{filepath}" -ar 16000 -ac 1 -c:a pcm_s16le "{audio_file}"')
     return audio_file
 
-pipe = pipeline("automatic-speech-recognition", model="NbAiLab/nb-whisper-large", chunk_length_s=30, device=device)
+pipe = pipeline("automatic-speech-recognition", model="NbAiLab/nb-whisper-large", chunk_length_s=30, torch_dtype=dtype).to(device)
 
 def transcribe_audio(audio_file, batch_size=16):
     start_time = time.time()