Hugging Face's logo

Spaces:
oza75
/
bambara-asr
Running on Zero

App Files Community
1
oza75 commited on
Commit
abdd451
•
1 Parent(s): affcb59

switch back to v002

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -12,10 +12,10 @@ from bambara_utils import BambaraWhisperTokenizer
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
 
14
  # Define the model checkpoint and language
15
- # model_checkpoint = "oza75/whisper-bambara-asr-002"
16
- # revision = "831cd15ed74a554caac9f304cf50dc773841ba1b"
17
- model_checkpoint = "oza75/whisper-bambara-asr-001"
18
- revision = "3578bcb14a42a5d2c58a436fb2c38341898e7885"
19
  language = "bambara"
20
 
21
  # Load the custom tokenizer designed for Bambara and the ASR model
 
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
 
14
  # Define the model checkpoint and language
15
+ model_checkpoint = "oza75/whisper-bambara-asr-002"
16
+ revision = "831cd15ed74a554caac9f304cf50dc773841ba1b"
17
+ # model_checkpoint = "oza75/whisper-bambara-asr-001"
18
+ # revision = "3578bcb14a42a5d2c58a436fb2c38341898e7885"
19
  language = "bambara"
20
 
21
  # Load the custom tokenizer designed for Bambara and the ASR model