Update app.py

app.py

@@ -14,6 +14,7 @@ model = WhisperForConditionalGeneration.from_pretrained(
 import torch
 
 processor = WhisperProcessor.from_pretrained('WajeehAzeemX/whisper-smal-ar-testing-kale-5000')
+forced_decoder_ids = processor.get_decoder_prompt_ids(language="arabic", task="transcribe")
 
 from transformers import GenerationConfig, WhisperForConditionalGeneration
 generation_config = GenerationConfig.from_pretrained("openai/whisper-small") # if you are using a multilingual model
@@ -43,7 +44,7 @@ async def transcribe_audio(request: Request):
         input_features = processor(audio_array, sampling_rate=sampling_rate, return_tensors="pt").input_features 
 
         # Generate token ids
-        predicted_ids = model.generate(input_features, return_timestamps=True)
+        predicted_ids = model.generate(input_features, forced_decoder_ids=forced_decoder_ids)
 
         # Decode token ids to text
         transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)