Update app.py

app.py

@@ -18,9 +18,9 @@ import torch
 auth_token = os.environ.get('TOKEN')
 
 
-M1 = "rohitp1/kkkh_whisper_small_distillation_att_loss_libri360_epochs_100_batch_4_concat_dataset"
-M2 = "rohitp1/dgx2_whisper_small_finetune_teacher_babble_noise_libri_360_hours_50_epochs_batch_8"
-M3 = "rohitp1/subhadeep_whisper_small_finetune_teacher_no_noise_libri_360_hours_100_epochs_batch_8"
+M1 = "rohitp1/subh_whisper_small_distil_att_loss_mozilla_epochs_50_batch_8"
+M2 = "rohitp1/dgx1_whisper_small_finetune_teacher_babble_noise_mozilla_40_epochs_batch_8"
+M3 = "openai/whisper-small.en"
 
 model1 = WhisperForConditionalGeneration.from_pretrained(M1, use_auth_token=auth_token)
 tokenizer1 = WhisperTokenizer.from_pretrained(M1, use_auth_token=auth_token)
@@ -57,7 +57,7 @@ def transcribe(mic_input, upl_input, model_type):
         text = p2(audio)["text"]
     elif model_type == 'CleanFinetuned':
         text = p3(audio)["text"]
-    elif model_type == 'NoisyDistillationQuantised':
+    elif model_type == 'DistilledQuantised':
         text = p1_quant(audio)['text']
     else:
         text = p1(audio)["text"]
@@ -115,7 +115,7 @@ if __name__ == "__main__":
                 )
 
                 with gr.Row():
-                    model_type = gr.inputs.Dropdown(["RobustDistillation", "NoisyFinetuned", "CleanFinetuned", "NoisyDistillationQuantised"], label='Model Type')
+                    model_type = gr.inputs.Dropdown(["RobustDistillation", "NoisyFinetuned", "CleanFinetuned", "DistilledAndQuantised"], label='Model Type')
 
                 with gr.Row():
                     clr_btn = gr.Button(value="Clear", variant="secondary")
@@ -152,7 +152,7 @@ if __name__ == "__main__":
                 """
                 <h4>Credits</h4>
                 Author:  Rohit Prasad <br>
-                Check out the model <a href="https://huggingface.co/rohitp1/kkkh_whisper_small_distillation_att_loss_libri360_epochs_100_batch_4_concat_dataset">here</a>
+                Check out the model <a href="https://huggingface.co/rohitp1/subh_whisper_small_distil_att_loss_mozilla_epochs_50_batch_8">here</a>
                 """
             )