import gradio as gr

from transformers import pipeline

pipe = pipeline("translation", model="facebook/nllb-200-distilled-600M")

def predict(text):
  return pipe(text,src_lang='eng_Latn',tgt_lang='zho_Hans')[0]['translation_text']

demo = gr.Interface(
  fn=predict,
  inputs='text',
  outputs='text',
)

demo.launch(share=True)