Update app.py

app.py

@@ -16,14 +16,13 @@ model_id = "WajeehAzeemX/whisper-smal-ar-testing-kale-5000"
 model = WhisperForConditionalGeneration.from_pretrained(
     model_id
 )
-processor = WhisperProcessor.from_pretrained(model_id, attn_implementation="sdpa")
-model.config.forced_decoder_ids = None
-forced_decoder_ids = processor.get_decoder_prompt_ids(language="Arabic", task="transcribe")
+processor = WhisperProcessor.from_pretrained(model_id)
 pipe = pipeline(
     "automatic-speech-recognition",
     model=model,
     tokenizer=processor.tokenizer,
     feature_extractor=processor.feature_extractor,
+    generate_kwargs = {"task":"transcribe", "language":"<|ar|>"}
 )
 
 
@@ -39,7 +38,7 @@ async def transcribe_audio(request: Request):
         # Process the audio array
         input_features = processor(audio_array, sampling_rate=sampling_rate, return_tensors="pt").input_features
         # Generate token ids
-        predicted_ids = model.generate(input_features, forced_decoder_ids=forced_decoder_ids)
+        predicted_ids = model.generate(input_features)
         # Decode token ids to text
         transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)
         # Print the transcription