Update app.py

app.py

@@ -44,15 +44,14 @@ def translate_text(text, destination_language):
     return translation[0]["translation_text"]
 
 # Initialize the speech-to-text pipeline (Whisper model)
-speech_to_text = pipeline(
-    "automatic-speech-recognition",
-    model="openai/whisper-small",
-    device=0  # Make sure to use the appropriate device (0 for GPU, or -1 for CPU)
-)
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+
+pipe = pipeline("text-to-audio", model="facebook/musicgen-small")
 
 # Function to transcribe audio to text
 def transcribe_audio(audio_file, destination_language):
-    transcription = speech_to_text(audio_file)["text"]
+    transcription = pipe(audio_file)["text"]
     return translate_text(transcription, destination_language)
 
 # Gradio interface