Update app.py

app.py

@@ -14,7 +14,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 whisper_model = pipeline(
     task="automatic-speech-recognition",
-    model="openai/whisper-large-v2",
+    model="openai/whisper-small",
     chunk_length_s=30,
     device=device,
 )
@@ -152,19 +152,22 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
         with gr.Row(elem_id="chat_row"):
             chatbot = gr.Chatbot(elem_id="chat_bot", visible=False).style(color_map=("green", "blue"))
             chatbot1 = gr.Chatbot(elem_id="chat_bot1").style(color_map=("green", "blue"))
-        with gr.Row(elem_id="prompt_row"):
-            prompt_input_audio = gr.Audio(label = 'Record Audio Input',source="microphone",type="filepath")
+        with gr.Row(elem_id="prompt_row"):    
+            with gr.Column():
+                prompt_input_audio = gr.Audio(
+                                        source="microphone",
+                                        type="filepath",
+                                        label="Record Audio Input",
+                                        
+                                    )
+                with gr.Row():
+                    translate_btn = gr.Button("Check Whisper first ? 👍")              
+                    translate_btn.click(fn=translate, 
+                                 inputs=prompt_input_audio, 
+                                 outputs=prompt_input
+                                )
             prompt_input = gr.Textbox(lines=2, label="Input text",show_label=True)
             chat_history = gr.Textbox(lines=4, label="prompt", visible=False)
-            translate_btn = gr.Button(value = "Translate").style(
-                    margin=True,
-                    rounded=(True, True, True, True),
-                    width=100
-                )
-            translate_btn.click(fn=translate, 
-                             inputs=prompt_input_audio, 
-                             outputs=prompt_input
-                            )
             submit_btn = gr.Button(value = "Submit",elem_id="submit-btn").style(
                     margin=True,
                     rounded=(True, True, True, True),