Update app.py

app.py

@@ -6,7 +6,7 @@
 import gradio as gr
 import os
 import transformers
-from transformers import pipeline, WhisperForConditionalGeneration
+from transformers import pipeline, WhisperForConditionalGeneration, WhisperTokenizer, WhisperFeatureExtractor
 import time
 
 # def greet_from_secret(ignored_param):
@@ -18,7 +18,13 @@ auth_token = os.environ.get('TOKEN')
 model = WhisperForConditionalGeneration.from_pretrained("rohitp1/kkkh_whisper_small_distillation_att_loss_libri360_epochs_100_batch_4_concat_dataset",
                                                          use_auth_token=auth_token)
 
-p = pipeline('automatic-speech-recognition', model=model)
+tokenizer = WhisperTokenizer.from_pretrained("rohitp1/kkkh_whisper_small_distillation_att_loss_libri360_epochs_100_batch_4_concat_dataset",
+                                                         use_auth_token=auth_token)
+
+feat_ext = WhisperFeatureExtractor.from_pretrained("rohitp1/kkkh_whisper_small_distillation_att_loss_libri360_epochs_100_batch_4_concat_dataset",
+                                                         use_auth_token=auth_token)
+
+p = pipeline('automatic-speech-recognition', model=model, tokenizer=tokenizer, feature_extractor=feat_ext)
 
 def transcribe(audio, state=""):
     time.sleep(3)