Spaces:

oza75
/

bambara-asr

Sleeping

oza75 commited on May 27

Commit

abdd451

•

1 Parent(s): affcb59

switch back to v002

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,10 +12,10 @@ 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"
 # Load the custom tokenizer designed for Bambara and the ASR model

 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"
 # Load the custom tokenizer designed for Bambara and the ASR model