import gradio as gr
from transformers import M2M100Tokenizer, AutoModelForSeq2SeqLM
def greet(input_text):
    tokenizer = M2M100Tokenizer.from_pretrained("CadenzaBaron/M2M100-418M-for-GameTranslation-Finetuned-Zh-En")
    model = AutoModelForSeq2SeqLM.from_pretrained("CadenzaBaron/M2M100-418M-for-GameTranslation-Finetuned-Zh-En")
    tokenizer.src_lang = "zh"
    tokenizer.tgt_lang = "en"
    inputs = tokenizer(input_text, return_tensors="pt")
    translated_tokens = model.generate(**inputs, num_beams=10, do_sample=True)
    translation = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
    return translation

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