Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
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"
|
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",
|