Update app.py

app.py

@@ -1,11 +1,14 @@
+import os
 import ctranslate2
 from transformers import AutoTokenizer
 import gradio as gr
 
-device = "cuda" if ctranslate2.get_cuda_device_count() > 0 else "cpu"
+# 自动 clone 模型（避免上传）
+if not os.path.exists("nllb_ct2"):
+    os.system("git lfs install && git clone https://huggingface.co/JustFrederik/nllb-200-distilled-600M-ct2-int8 nllb_ct2")
 
-# 使用公开模型 JustFrederik/nllb-200-distilled-600M-ct2-int8
-translator = ctranslate2.Translator("JustFrederik/nllb-200-distilled-600M-ct2-int8", device=device)
+device = "cuda" if ctranslate2.get_cuda_device_count() > 0 else "cpu"
+translator = ctranslate2.Translator("nllb_ct2", device=device)
 tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
 
 def translate(text, source_lang="eng_Latn", target_lang="zho_Hans"):
@@ -23,5 +26,5 @@ gr.Interface(
         gr.Textbox(value="zho_Hans", label="目标语言代码"),
     ],
     outputs="text",
-    title="NLLB-200 翻译器（基于 CTranslate2）",
+    title="NLLB-200 翻译器（自动挂载模型）",
 ).launch()