Update app.py

app.py

@@ -6,7 +6,7 @@
 import gradio as gr
 import os
 import transformers
-from transformers import pipeline, WhisperFromConditionalGeneration
+from transformers import pipeline, WhisperForConditionalGeneration
 import time
 
 # def greet_from_secret(ignored_param):
@@ -15,7 +15,7 @@ import time
 
 
 auth_token = os.environ.get('TOKEN')
-model = WhisperFromConditionalGeneration.from_pretrained("rohitp1/kkkh_whisper_small_distillation_att_loss_libri360_epochs_100_batch_4_concat_dataset",
+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)