Update app.py

app.py

@@ -23,7 +23,7 @@ from pydub import AudioSegment
 import torch
 import torchaudio
 import torchaudio.transforms as transforms
-from transformers import WhisperTokenizer, WhisperForConditionalGeneration, WhisperProcessor
+from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
 from huggingface_hub import model_info
 import spacy
 import networkx as nx
@@ -52,9 +52,8 @@ device = 0 if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float32
 
 # Load tokenizer and model
-tokenizer = WhisperTokenizer.from_pretrained("NbAiLab/nb-whisper-large")
-model = WhisperForConditionalGeneration.from_pretrained("NbAiLab/nb-whisper-large")
-processor = WhisperProcessor.from_pretrained("NbAiLab/nb-whisper-large")
+processor = AutoProcessor.from_pretrained("NbAiLab/nb-whisper-large-verbatim")
+model = AutoModelForSpeechSeq2Seq.from_pretrained("NbAiLab/nb-whisper-large-verbatim")
 
 
 # Model script does not support JIT compilation