Update app.py

app.py

@@ -6,7 +6,7 @@ import torch
 import gradio as gr
 import pytube as pt
 import spaces
-from transformers import AutoTokenizer, WhisperForConditionalGeneration, WhisperProcessor, pipeline
+from transformers import AutoFeatureExtractor, AutoTokenizer, WhisperForConditionalGeneration, WhisperProcessor, pipeline
 from huggingface_hub import model_info
 try:
     import flash_attn
@@ -34,6 +34,7 @@ def pipe(file, return_timestamps=False):
         task="automatic-speech-recognition",
         model=model,
         tokenizer=AutoTokenizer.from_pretrained(MODEL_NAME),
+        feature_extractor=AutoFeatureExtractor.from_pretrained(MODEL_NAME),
         chunk_length_s=30,
         device=device,
         token=auth_token,