oza75 commited on
Commit
6dcf9ee
·
verified ·
1 Parent(s): cd8d796

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -15,16 +15,16 @@ 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"
23
- #revision = "5f143f6070b64412a44fea08e912e1b7312e9ae9" # this checkpoint support both task without overfitting
24
 
25
- model_checkpoint = "oza75/whisper-bambara-asr-006"
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.
 
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"
23
+ revision = "5f143f6070b64412a44fea08e912e1b7312e9ae9" # this checkpoint support both task without overfitting
24
 
25
+ #model_checkpoint = "oza75/whisper-bambara-asr-006"
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.