Attempted to append the respond with a translation dataset for Vietnamese reponses.

app.py

@@ -1,14 +1,15 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-
 from datasets import load_dataset
 
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 dataset = load_dataset("JustKiddo/KiddosVault")
+translation = load_dataset("IWSLT/mt_eng_vietnamese")
 
+def translate_text(text, translation_dataset):
+    # Assuming the translation dataset has a method to translate text
+    translated_text = translation_dataset['train'][0]['translation']['vi']  # Example translation
+    return translated_text
 
 def respond(
     message,
@@ -38,19 +39,18 @@ def respond(
         top_p=top_p,
     ):
         token = message.choices[0].delta.content
-
         response += token
         yield response
 
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
+    translated_response = translate_text(response, translation)
+    yield response + "\n\nTranslation: " + translated_response
+
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
         gr.Textbox(value="You are a professional Mental Healthcare Chatbot.", label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=2048, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=1, step=0.1, label="Temperature"),
+        gr.Slider(minimum=0.1, maximum 4.0, value=1, step=0.1, label="Temperature"),
         gr.Slider(
             minimum=0.1,
             maximum=1.0,
@@ -61,6 +61,5 @@ demo = gr.ChatInterface(
     ],
 )
 
-
 if __name__ == "__main__":
     demo.launch()