rohitp1 commited on
Commit
5cb989a
·
1 Parent(s): a3d358b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ auth_token = os.environ.get('TOKEN')
20
 
21
  M1 = "rohitp1/subh_whisper_small_distil_att_loss_mozilla_epochs_50_batch_8"
22
  M2 = "rohitp1/dgx1_whisper_small_finetune_teacher_babble_noise_mozilla_40_epochs_batch_8"
23
- M3 = "openai/whisper-small.en"
24
 
25
  model1 = WhisperForConditionalGeneration.from_pretrained(M1, use_auth_token=auth_token)
26
  tokenizer1 = WhisperTokenizer.from_pretrained(M1, use_auth_token=auth_token)
 
20
 
21
  M1 = "rohitp1/subh_whisper_small_distil_att_loss_mozilla_epochs_50_batch_8"
22
  M2 = "rohitp1/dgx1_whisper_small_finetune_teacher_babble_noise_mozilla_40_epochs_batch_8"
23
+ M3 = "rohitp1/dgx1_whisper_small_finetune_teacher_no_noise_mozilla_40_epochs_batch_8"
24
 
25
  model1 = WhisperForConditionalGeneration.from_pretrained(M1, use_auth_token=auth_token)
26
  tokenizer1 = WhisperTokenizer.from_pretrained(M1, use_auth_token=auth_token)