added translation

app.py

@@ -24,14 +24,29 @@ import base64
 with open("Iso_Logotipo_Ceibal.png", "rb") as image_file:
     encoded_image = base64.b64encode(image_file.read()).decode()
 
-# CKPT = "facebook/nllb-200-distilled-600M"
+CKPT = "facebook/nllb-200-distilled-600M"
+
+model = AutoModelForSeq2SeqLM.from_pretrained(CKPT)
+tokenizer = AutoTokenizer.from_pretrained(CKPT)
+
+def translate_text(text):
+    translation_pipeline = pipeline("translation",
+                                    model=model,
+                                    tokenizer=tokenizer,
+                                    src_lang="spa_Latn",
+                                    tgt_lang="eng_Latn",
+                                    max_length=400,
+                                    device=device)
+
+    result = translation_pipeline(text)
+    return result[0]['translation_text']
+
+
 
-# model = AutoModelForSeq2SeqLM.from_pretrained(CKPT)
-# tokenizer = AutoTokenizer.from_pretrained(CKPT)
 def generate_sound(text,steps,audio_length,negative_prompt): 
     print(text)
-    # text=translate_text(text)
-    # negative_prompt = translate_text(negative_prompt)
+    text=translate_text(text)
+    negative_prompt = translate_text(negative_prompt)
     print(text)
     waveforms = pipe(text, 
                      num_inference_steps=steps, 
@@ -40,17 +55,7 @@ def generate_sound(text,steps,audio_length,negative_prompt):
     rate =16000
     return rate, waveforms[0]
 
-# def translate_text(text):
-#     translation_pipeline = pipeline("translation",
-#                                     model=model,
-#                                     tokenizer=tokenizer,
-#                                     src_lang="spa_Latn",
-#                                     tgt_lang="eng_Latn",
-#                                     max_length=400,
-#                                     device=device)
 
-#     result = translation_pipeline(text)
-#     return result[0]['translation_text']
 # def translate_text(text):
 #     text = es_en_translator(text)[0].get("translation_text")
 #     return text