Update app.py

app.py

@@ -52,6 +52,17 @@ LANGUANGE_MAP = {
  }
 
 
+processor = WhisperProcessor.from_pretrained(model_id)
+model = WhisperForConditionalGeneration.from_pretrained(model_id)
+model.eval()
+model.to(device)
+
+sampling_rate = processor.feature_extractor.sampling_rate
+
+bos_token_id = processor.tokenizer.all_special_ids[-106]
+decoder_input_ids = torch.tensor([bos_token_id]).to(device)
+
+
 device = "cuda" if torch.cuda.is_available() else "CPU"
 
 model_ckpt = "barto17/language-detection-fine-tuned-on-xlm-roberta-base"