oza75 commited on
Commit
b13531b
·
verified ·
1 Parent(s): 2c64e2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -16,11 +16,11 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
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"
@@ -102,7 +102,7 @@ def main():
102
  fn=transcribe,
103
  inputs=[
104
  gr.Audio(type="filepath", value=example_files[0][0]),
105
- gr.Radio(choices=["transcribe", "translate"], label="Task Type", value="transcribe")
106
  ],
107
  outputs="text",
108
  title="Bambara Automatic Speech Recognition",
 
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"
 
102
  fn=transcribe,
103
  inputs=[
104
  gr.Audio(type="filepath", value=example_files[0][0]),
105
+ gr.Radio(choices=["transcribe"], label="Task Type", value="transcribe")
106
  ],
107
  outputs="text",
108
  title="Bambara Automatic Speech Recognition",