Update app.py

app.py

@@ -13,8 +13,19 @@ app = FastAPI()
 # Load the Whisper model once during startup
 device = 0 if torch.cuda.is_available() else -1  # Use GPU if available, otherwise CPU
 # asr_pipeline = pipeline(model="openai/whisper-small", device=device)  # Initialize Whisper model
+# asr_pipeline = pipeline( model="openai/whisper-small", device=device, language="pt")
+
+# Load model and processor
+model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-small")
+processor = WhisperProcessor.from_pretrained("openai/whisper-small")
+
+# Set forced language to Portuguese (pt)
+forced_language_token_id = processor.tokenizer.convert_tokens_to_ids("<|pt|>")
+model.config.forced_decoder_ids = [[2, forced_language_token_id]]  # `2` refers to the decoder start token.
+
+# Initialize the pipeline
+asr_pipeline = pipeline(task="automatic-speech-recognition", model=model, processor=processor, device=device)
 
-asr_pipeline = pipeline( model="openai/whisper-small", device=device, language="pt")
 
 # Basic GET endpoint
 @app.get("/")