Update app.py

app.py

@@ -3,22 +3,22 @@ import warnings
 import torch
 from transformers import WhisperTokenizer, WhisperForConditionalGeneration, WhisperProcessor
 import soundfile as sf
-from huggingface_hub import login
 
 warnings.filterwarnings("ignore")
 
-
-# tokenizer, model
+# Load tokenizer + model
 tokenizer = WhisperTokenizer.from_pretrained("NbAiLabBeta/nb-whisper-medium")
 model = WhisperForConditionalGeneration.from_pretrained("NbAiLabBeta/nb-whisper-medium")
 processor = WhisperProcessor.from_pretrained("NbAiLabBeta/nb-whisper-medium")
 
-# Set up device
-device = ('cuda')
-# device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+# set up device
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 torch_dtype = torch.float32
+
+# move model to device
 model.to(device)
 
+
 def transcribe_audio(audio_file):
     audio_input, _ = sf.read(audio_file)
     inputs = processor(audio_input, sampling_rate=16000, return_tensors="pt")
@@ -41,7 +41,7 @@ banner_html = """
 </div>
 """
 
-# Create Gradio interface
+# Gradio interface
 iface = gr.Blocks()
 
 with iface: