oza75 commited on
Commit
3bff29a
·
verified ·
1 Parent(s): b13531b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -15,8 +15,9 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
15
  # Define the model checkpoint and language
16
  #model_checkpoint = "oza75/whisper-bambara-asr-002"
17
  #revision = "831cd15ed74a554caac9f304cf50dc773841ba1b"
18
- model_checkpoint = "oza75/whisper-bambara-asr-005"
19
- revision = "6a92cd0f19985d12739c2f6864607627115e015d" # first good checkpoint for bambara
 
20
  #revision = "fb69a5750182933868397543366dbb63747cf40c" # this only translate in english
21
  #revision = "129f9e68ead6cc854e7754b737b93aa78e0e61e1" # support transcription and translation
22
  #revision = "cb8e351b35d6dc524066679d9646f4a947300b27"
@@ -26,6 +27,9 @@ revision = "6a92cd0f19985d12739c2f6864607627115e015d" # first good checkpoint fo
26
  #revision = "96535debb4ce0b7af7c9c186d09d088825f63840"
27
  #revision = "4549778c08f29ed2e033cc9a497a187488b6bf56"
28
 
 
 
 
29
  # language = "bambara"
30
  language = "icelandic" # we use icelandic as the model was trained to replace the icelandic with bambara.
31
 
 
15
  # Define the model checkpoint and language
16
  #model_checkpoint = "oza75/whisper-bambara-asr-002"
17
  #revision = "831cd15ed74a554caac9f304cf50dc773841ba1b"
18
+ # model_checkpoint = "oza75/whisper-bambara-asr-005"
19
+ # revision = "6a92cd0f19985d12739c2f6864607627115e015d" # first good checkpoint for bambara
20
+
21
  #revision = "fb69a5750182933868397543366dbb63747cf40c" # this only translate in english
22
  #revision = "129f9e68ead6cc854e7754b737b93aa78e0e61e1" # support transcription and translation
23
  #revision = "cb8e351b35d6dc524066679d9646f4a947300b27"
 
27
  #revision = "96535debb4ce0b7af7c9c186d09d088825f63840"
28
  #revision = "4549778c08f29ed2e033cc9a497a187488b6bf56"
29
 
30
+ model_checkpoint = "oza75/bm-whisper-02"
31
+ revision = "06f14ac7038bc75130ea8ec4a3e0ef2b43a7609f"
32
+
33
  # language = "bambara"
34
  language = "icelandic" # we use icelandic as the model was trained to replace the icelandic with bambara.
35