Update app.py
Browse files
app.py
CHANGED
|
@@ -12,14 +12,11 @@ MODELS = {
|
|
| 12 |
"Soloba V3 (CTC)": ("RobotsMali/soloba-ctc-0.6b-v3", "ctc"),
|
| 13 |
"Soloba V2 (CTC)": ("RobotsMali/soloba-ctc-0.6b-v2", "ctc"),
|
| 14 |
"Soloba V1 (CTC)": ("RobotsMali/soloba-ctc-0.6b-v1", "ctc"),
|
| 15 |
-
"Soloba V0 (CTC)": ("RobotsMali/soloba-ctc-0.6b-v0", "ctc"),
|
| 16 |
"Soloba V1.5 (TDT)": ("RobotsMali/soloba-tdt-0.6b-v1.5", "rnnt"),
|
| 17 |
"Soloba V0.5 (TDT)": ("RobotsMali/soloba-tdt-0.6b-v0.5", "rnnt"),
|
| 18 |
"Soloni V3 (TDT-CTC)": ("RobotsMali/soloni-114m-tdt-ctc-v3", "rnnt"),
|
| 19 |
"Soloni V2 (TDT-CTC)": ("RobotsMali/soloni-114m-tdt-ctc-v2", "rnnt"),
|
| 20 |
"Soloni V1 (TDT-CTC)": ("RobotsMali/soloni-114m-tdt-ctc-v1", "rnnt"),
|
| 21 |
-
"Soloni V0 (TDT-CTC)": ("RobotsMali/soloni-114m-tdt-ctc-v0", "rnnt"),
|
| 22 |
-
"Soloni MSE (Experimental)": ("RobotsMali/lau-soloni-114m-mse-k1", "ctc"),
|
| 23 |
"Traduction Soloni (ST)": ("RobotsMali/st-soloni-114m-tdt-ctc", "rnnt"),
|
| 24 |
}
|
| 25 |
|
|
@@ -129,7 +126,7 @@ def pipeline(video_in, model_name):
|
|
| 129 |
|
| 130 |
# 5. INTERFACE GRADIO
|
| 131 |
with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
| 132 |
-
gr.HTML("<div style='text-align:center;'><h1>🤖 RobotsMali Speech
|
| 133 |
|
| 134 |
with gr.Row():
|
| 135 |
with gr.Column():
|
|
@@ -138,7 +135,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
| 138 |
run_btn = gr.Button("🚀 GÉNÉRER", variant="primary")
|
| 139 |
|
| 140 |
if EXAMPLE_PATH:
|
| 141 |
-
gr.Examples(examples=[[EXAMPLE_PATH, "
|
| 142 |
|
| 143 |
with gr.Column():
|
| 144 |
status = gr.Markdown("### État\nEn attente...")
|
|
|
|
| 12 |
"Soloba V3 (CTC)": ("RobotsMali/soloba-ctc-0.6b-v3", "ctc"),
|
| 13 |
"Soloba V2 (CTC)": ("RobotsMali/soloba-ctc-0.6b-v2", "ctc"),
|
| 14 |
"Soloba V1 (CTC)": ("RobotsMali/soloba-ctc-0.6b-v1", "ctc"),
|
|
|
|
| 15 |
"Soloba V1.5 (TDT)": ("RobotsMali/soloba-tdt-0.6b-v1.5", "rnnt"),
|
| 16 |
"Soloba V0.5 (TDT)": ("RobotsMali/soloba-tdt-0.6b-v0.5", "rnnt"),
|
| 17 |
"Soloni V3 (TDT-CTC)": ("RobotsMali/soloni-114m-tdt-ctc-v3", "rnnt"),
|
| 18 |
"Soloni V2 (TDT-CTC)": ("RobotsMali/soloni-114m-tdt-ctc-v2", "rnnt"),
|
| 19 |
"Soloni V1 (TDT-CTC)": ("RobotsMali/soloni-114m-tdt-ctc-v1", "rnnt"),
|
|
|
|
|
|
|
| 20 |
"Traduction Soloni (ST)": ("RobotsMali/st-soloni-114m-tdt-ctc", "rnnt"),
|
| 21 |
}
|
| 22 |
|
|
|
|
| 126 |
|
| 127 |
# 5. INTERFACE GRADIO
|
| 128 |
with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
| 129 |
+
gr.HTML("<div style='text-align:center;'><h1>🤖 RobotsMali Speech Lab</h1></div>")
|
| 130 |
|
| 131 |
with gr.Row():
|
| 132 |
with gr.Column():
|
|
|
|
| 135 |
run_btn = gr.Button("🚀 GÉNÉRER", variant="primary")
|
| 136 |
|
| 137 |
if EXAMPLE_PATH:
|
| 138 |
+
gr.Examples(examples=[[EXAMPLE_PATH, "Soloni V3 (TDT-CTC)"]], inputs=[v_input, m_input])
|
| 139 |
|
| 140 |
with gr.Column():
|
| 141 |
status = gr.Markdown("### État\nEn attente...")
|