Update app.py

app.py

@@ -30,7 +30,7 @@ processor = AutoProcessor.from_pretrained("NbAiLabBeta/nb-whisper-large")
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 torch_dtype = torch.float32
 
-pipe = pipeline("automatic-speech-recognition", model="NbAiLabBeta/nb-whisper-large", device=device, torch_dtype=torch_dtype)
+pipe = pipeline("automatic-speech-recognition", model="NbAiLabBeta/nb-whisper-large", torch_dtype=torch_dtype)
 
 language = "no"
 task = "transcribe"
@@ -47,7 +47,7 @@ def transcribe_audio(audio_file):
 
     with torch.no_grad():
         # CUDA within the function
-        with torch.cuda.device(device) if torch.cuda.is_available() else contextlib.nullcontext():
+        # with torch.cuda.device(device) if torch.cuda.is_available() else contextlib.nullcontext():
             output = pipe(audio_file, chunk_length_s=30, generate_kwargs={"forced_decoder_ids": forced_decoder_ids})
 
     text = output["text"]
@@ -185,7 +185,7 @@ def text_rank_summary(text, num_paragraphs=3):
 
 banner_html = """
 <div style="text-align: center;">
-    <img src="https://huggingface.co/spaces/camparchimedes/transcription_app/blob/main/banner_trans.png" alt="" width="100%" height="auto">
+    <img src="https://github.com/camparchimedes/sw-llm/blob/main/annex/cooltext462376124862020.png" alt="" width="100%" height="auto">
 </div>
 """