Update app.py

app.py

@@ -54,7 +54,7 @@ SIDEBAR_INFO = f"""
     <img src="{LOGO}" style="width: 100%; height: auto;"/>
 </div>
 """
-device = 0 if torch.cuda.is_available() else -1
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 def convert_to_wav(filepath):
     _,file_ending = os.path.splitext(f'{filepath}')
@@ -62,12 +62,12 @@ 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", device=device)
+pipe = pipeline("automatic-speech-recognition", model="NbAiLab/nb-whisper-large", chunk_length_s=30, device=device)
 
-def transcribe_audio(audio_file):
+def transcribe_audio(audio_file, batch_size=16):
     start_time = time.time()
 
-    outputs = pipe(audio_file, return_timestamps=False, generate_kwargs={'task': 'transcribe', 'language': 'no'}) # skip_special_tokens=True
+    outputs = pipe(audio_file, batch_size=batch_size, return_timestamps=False, generate_kwargs={'task': 'transcribe', 'language': 'no'}) # skip_special_tokens=True
     text = outputs["text"]
 
     end_time = time.time()