Update app.py

app.py

@@ -4,15 +4,14 @@ from whisper import generate
 from AinaTheme import theme
 
 MODEL_NAME = "openai/whisper-large-v3"
-USE_V4 = False
 
-
-def transcribe(inputs, use_v4):
+def transcribe(inputs, model_version):
     if inputs is None:
         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")
 
-    return generate(audio_path=inputs, use_v4=use_v4)
+    usev4 = model_version=="v0.4"
+    return generate(audio_path=inputs, use_v4=usev4)
 
 
 description_string = "Transcripció automàtica de micròfon o de fitxers d'àudio.\n Aquest demostrador s'ha desenvolupat per"\
@@ -23,7 +22,7 @@ description_string = "Transcripció automàtica de micròfon o de fitxers d'àud
 def clear():
      return (
           None,
-          USE_V4
+          "v0.3"
      )
 
 
@@ -31,7 +30,7 @@ with gr.Blocks(theme=theme) as demo:
     gr.Markdown(description_string)
     with gr.Row():
         with gr.Column(scale=1):
-            use_v4 = gr.Checkbox(label="Use v4", value=USE_V4)
+            model_version = gr.Dropdown(label="Model Version", choices=["v0.3", "v0.4"], value="v0.3")
             input = gr.Audio(sources=["upload", "microphone"], type="filepath", label="Audio")
 
         with gr.Column(scale=1):
@@ -42,8 +41,8 @@ with gr.Blocks(theme=theme) as demo:
             submit_btn = gr.Button("Submit", variant="primary")
 
 
-    submit_btn.click(fn=transcribe, inputs=[input, use_v4], outputs=[output])
-    clear_btn.click(fn=clear,inputs=[], outputs=[input, use_v4], queue=False,)
+    submit_btn.click(fn=transcribe, inputs=[input, model_version], outputs=[output])
+    clear_btn.click(fn=clear,inputs=[], outputs=[input, model_version], queue=False,)
 
 
 if __name__ == "__main__":