Update to whisper

app.py

@@ -8,7 +8,7 @@ import plotly.express as px
 
 import torch
 import gradio as gr
-from transformers import pipeline, Wav2Vec2ProcessorWithLM
+from transformers import pipeline, WhisperProcessor
 from pyannote.audio import Pipeline
 from librosa import load, resample
 from rpunct import RestorePuncts
@@ -64,8 +64,8 @@ def summarize(diarized, check):
 
 
 # Audio components
-asr_model = "patrickvonplaten/wav2vec2-base-960h-4-gram"
-processor = Wav2Vec2ProcessorWithLM.from_pretrained(asr_model)
+asr_model = "openai/whisper-large"
+processor = WhisperProcessor.from_pretrained(asr_model)
 asr = pipeline(
     "automatic-speech-recognition",
     model=asr_model,