Update app.py (#3)

- Update app.py (d86317defc7b35eb52a5f144c3c524b219125712)


Co-authored-by: Sarah Solito <[email protected]>

app.py

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