Update app.py

app.py

@@ -23,7 +23,7 @@ all_special_ids = whisper_model.tokenizer.all_special_ids
 transcribe_token_id = all_special_ids[-5]
 translate_token_id = all_special_ids[-6]
 
-def transcribe(audio):
+def translate(audio):
     task = "translate"
     whisper_model.model.config.forced_decoder_ids = [[2, transcribe_token_id if task=="transcribe" else translate_token_id]]
     text = whisper_model(audio)["text"]
@@ -156,12 +156,12 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
             prompt_input_audio = gr.Audio(label = 'Record Audio Input',source="microphone",type="filepath")
             prompt_input = gr.Textbox(lines=2, label="Input text",show_label=True)
             chat_history = gr.Textbox(lines=4, label="prompt", visible=False)
-            transcribe_btn = gr.Button(value = "Transcribe").style(
+            translate_btn = gr.Button(value = "Translate").style(
                     margin=True,
                     rounded=(True, True, True, True),
                     width=100
                 )
-            transcribe_btn.click(fn=transcribe, 
+            translate_btn.click(fn=translate, 
                              inputs=prompt_input_audio, 
                              outputs=prompt_input
                             )