Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,7 +3,7 @@ from whisper import generate
3
  from AinaTheme import theme
4
  import spaces
5
 
6
- USE_V5 = False
7
 
8
  @spaces.GPU
9
  def transcribe(inputs, model_version):
@@ -11,8 +11,8 @@ def transcribe(inputs, model_version):
11
  raise gr.Error("Cap fitxer d'àudio introduit! Si us plau pengeu un fitxer "\
12
  "o enregistreu un àudio abans d'enviar la vostra sol·licitud")
13
 
14
- use_v5 = model_version == "v0.5"
15
- return generate(audio_path=inputs, use_v5=use_v5)
16
 
17
  description_string = "Transcripció automàtica de micròfon o de fitxers d'àudio.\n Aquest demostrador s'ha desenvolupat per"\
18
  " comprovar els models de reconeixement de parla per a móbils."
@@ -24,7 +24,7 @@ with gr.Blocks() as demo:
24
  gr.Markdown(description_string)
25
  with gr.Row():
26
  with gr.Column(scale=1):
27
- model_version = gr.Dropdown(label="Model Version", choices=["v1.0"], value="v1.0")
28
  input = gr.Audio(sources=["upload", "microphone"], type="filepath", label="Audio")
29
 
30
  with gr.Column(scale=1):
 
3
  from AinaTheme import theme
4
  import spaces
5
 
6
+ USE_V2 = False
7
 
8
  @spaces.GPU
9
  def transcribe(inputs, model_version):
 
11
  raise gr.Error("Cap fitxer d'àudio introduit! Si us plau pengeu un fitxer "\
12
  "o enregistreu un àudio abans d'enviar la vostra sol·licitud")
13
 
14
+ use_v2 = model_version == "v2.0"
15
+ return generate(audio_path=inputs, use_v2=use_v2)
16
 
17
  description_string = "Transcripció automàtica de micròfon o de fitxers d'àudio.\n Aquest demostrador s'ha desenvolupat per"\
18
  " comprovar els models de reconeixement de parla per a móbils."
 
24
  gr.Markdown(description_string)
25
  with gr.Row():
26
  with gr.Column(scale=1):
27
+ model_version = gr.Dropdown(label="Model Version", choices=["v1.0, v2.0"], value="v1.0")
28
  input = gr.Audio(sources=["upload", "microphone"], type="filepath", label="Audio")
29
 
30
  with gr.Column(scale=1):