oza75 commited on
Commit
b1b5d1d
·
verified ·
1 Parent(s): 964c1a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -33,9 +33,12 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
33
  # language = "bambara"
34
  # language = "icelandic" # we use icelandic as the model was trained to replace the icelandic with bambara.
35
 
36
- model_checkpoint = "oza75/bm-whisper-from-swa-02"
37
  revision = None
38
- language = "swahili"
 
 
 
39
 
40
  # Load the custom tokenizer designed for Bambara and the ASR model
41
  #tokenizer = BambaraWhisperTokenizer.from_pretrained(model_checkpoint, language=language, device=device)
 
33
  # language = "bambara"
34
  # language = "icelandic" # we use icelandic as the model was trained to replace the icelandic with bambara.
35
 
36
+ #model_checkpoint = "oza75/bm-whisper-from-swa-02"
37
  revision = None
38
+ #language = "swahili"
39
+
40
+ model_checkpoit = "oza75/bm-whisper-large-v3-turbo"
41
+ language = "icelandic"
42
 
43
  # Load the custom tokenizer designed for Bambara and the ASR model
44
  #tokenizer = BambaraWhisperTokenizer.from_pretrained(model_checkpoint, language=language, device=device)