Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import yt_dlp as youtube_dl
-from transformers import pipeline, BitsAndBytesConfig, WhisperForConditionalGeneration, WhisperTokenizer
+from transformers import pipeline, BitsAndBytesConfig, WhisperForConditionalGeneration
 from transformers.pipelines.audio_utils import ffmpeg_read
 import torch
 from huggingface_hub import CommitScheduler
@@ -25,7 +25,7 @@ device = 0 if torch.cuda.is_available() else "cpu"
 bnb_config = BitsAndBytesConfig(load_in_4bit=True)
 
 # Load the model
-model = model_class.from_pretrained(model_name, config=bnb_config)
+model = WhisperForConditionalGeneration.from_pretrained(model_name, config=bnb_config)
 
 # bnb_config = bnb.QuantizationConfig(bits=4)
 pipe = pipeline(task="automatic-speech-recognition", model=model, chunk_length_s=30, device=device)