Update app.py

app.py

@@ -1,18 +1,13 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import pipeline
 
+pipe = pipeline(task="translation", model="facebook/nllb-200-distilled-600M")
 
-# 加载模型和处理器
-model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
 
 def greet(text,src_lang,tgt_lang):
     # 输入文本
-    tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M",src_lang=src_lang ,tgt_lang=tgt_lang, return_tensors="pt")
-    # 生成翻译的文本
-    input_ids = tokenizer.encode(text, return_tensors="pt")
-    outputs = model.generate(input_ids, )
-    decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return decoded
+    result=pipe(text,src_lang=src_lang ,tgt_lang=tgt_lang)
+    return result[0]["translation_text"]
 
 text = gr.Textbox(lines=5,label="输入文本",placeholder="请输入文本")
 src_lang = gr.Dropdown(label="源语言",choices=["afr_Latn",