Update app.py

app.py

@@ -12,15 +12,17 @@ from bambara_utils import BambaraWhisperTokenizer
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Define the model checkpoint and language
-model_checkpoint = "oza75/whisper-bambara-asr-002"
-revision = "831cd15ed74a554caac9f304cf50dc773841ba1b"
-# model_checkpoint = "oza75/whisper-bambara-asr-001"
-# revision = "3578bcb14a42a5d2c58a436fb2c38341898e7885"
-language = "bambara"
+#model_checkpoint = "oza75/whisper-bambara-asr-002"
+#revision = "831cd15ed74a554caac9f304cf50dc773841ba1b"
+model_checkpoint = "oza75/whisper-bambara-asr-004"
+revision = "592afa6d2e60bf7b2f2556e14852c2fadd3bc106"
+# language = "bambara"
+language = "icelandic"
 
 
 # Load the custom tokenizer designed for Bambara and the ASR model
-tokenizer = BambaraWhisperTokenizer.from_pretrained(model_checkpoint, language=language, device=device)
+#tokenizer = BambaraWhisperTokenizer.from_pretrained(model_checkpoint, language=language, device=device)
+tokenizer = WhisperTokenizer.from_pretrained(model_checkpoint, language=language, device=device)
 pipe = pipeline(model=model_checkpoint, tokenizer=tokenizer, device=device, revision=revision)