Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, WhisperForConditi
 model_id = "openai/whisper-medium"
 processor = WhisperProcessor.from_pretrained(model_id)
 
-model = WhisperModel.from_pretrained(model_id)
+model = WhisperForConditionalGeneration.from_pretrained(model_id)
 model.config.forced_decoder_ids = None
 
 # Define a function that takes an audio input and returns a transcription