import torch
from seamless_communication.inference import Translator


# Initialize a Translator object with a multitask model, vocoder on the GPU.
translator = Translator(
    "seamlessM4T_v2_large", "vocoder_v2", torch.device("cuda:0"), torch.float16
)


def translate_text(text):
    print("test")
    # text_output, speech_output = translator.predict(
    #     input=text,
    #     task_str="T2ST",
    #     tgt_lang="spa",
    #     src_lang="eng",
    #     text_generation_opts=None,
    #     unit_generation_opts=None,
    # )
    # print(text_output)