Update app.py

app.py

@@ -17,8 +17,8 @@ def convert_to_wav(audio_file):
 
 import torch
 from transformers import pipeline
-
-pipe = pipeline("automatic-speech-recognition", model="NbAiLab/nb-whisper-large")
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = pipeline("automatic-speech-recognition", model="NbAiLab/nb-whisper-large", device=device)
 
 def transcribe_audio(audio_file):
     if audio_file.endswith(".m4a"):
@@ -26,8 +26,8 @@ def transcribe_audio(audio_file):
 
     start_time = time.time()
 
-    with torch.no_grad():
-        output = pipe(audio_file)
+   # with torch.no_grad():
+    output = pipe(audio_file)
 
     text = output["text"]
     end_time = time.time()