binaryMao commited on
Commit
eb25311
·
verified ·
1 Parent(s): 24a54d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
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 Laboratory</h1></div>")
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, "Soloba V3 (CTC)"]], inputs=[v_input, m_input])
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...")