Update app.py

app.py

@@ -42,12 +42,12 @@ def translate_text(text, destination_language):
     
     translation = text_translator(text, src_lang="eng_Latn", tgt_lang=dest_code)
     return translation[0]["translation_text"]
-from transformers import pipeline
 
-# Initialize the speech-to-text pipeline
+# Initialize the speech-to-text pipeline (Whisper model)
 speech_to_text = pipeline(
     "automatic-speech-recognition",
-    model="openai/whisper-small"
+    model="openai/whisper-small",
+    device=0  # Make sure to use the appropriate device (0 for GPU, or -1 for CPU)
 )
 
 # Function to transcribe audio to text
@@ -55,7 +55,6 @@ def transcribe_audio(audio_file, destination_language):
     transcription = speech_to_text(audio_file)["text"]
     return translate_text(transcription, destination_language)
 
-
 # Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# AI-Powered Language Translation Bot")