add src_lang and tgt_lang

app.py

@@ -4,7 +4,8 @@ from huggingface_hub import InferenceClient
 import gradio as gr
 
 # Initialize the translation pipeline
-translation_pipeline = pipeline("translation", model="facebook/nllb-200-distilled-600M")
+# Specify the source and target languages for translation
+translation_pipeline = pipeline("translation", model="facebook/nllb-200-distilled-600M", src_lang="zh", tgt_lang="en")
 
 # Initialize the text-to-speech client
 # Make sure you have HF_TOKEN set as a secret in your Hugging Face Space
@@ -18,7 +19,8 @@ def translate_and_speak(chinese_text):
     Translates Chinese text to English and generates speech.
     """
     # Translate the text
-    translated_text = translation_pipeline(chinese_text)[0]['translation_text']
+    # Pass the input text as a list to the pipeline
+    translated_text = translation_pipeline([chinese_text])[0]['translation_text']
 
     # Generate speech from the translated text
     # Use a try-except block to handle potential errors during speech generation