import gradio as gr


from transformers.utils import logging
loggging.set_verbosity_error()


from transformers import pipeline
import torch 

translator = pipeline(task='translation', 
                     model='facebook/nllb-200-distilled-600M',
                         torch.dtype=torch.bfloat16)


def translate(text):
    translated_text = translator(text. src_lang='eng_Latn', tgt_lang='fra_Latn')
    return translated_text

demo = gr.Interface(fn=greet, inputs="text", outputs="text")
demo.launch()