Update app.py

app.py

@@ -54,14 +54,17 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 def prompting(text, source_lang, target_lang):
     l = LANGUAGES[target_lang]
     prompt=f"Translate the following {source_lang} sentence into {target_lang} and explain it in detail:\n{text} <{l}>"
+    return prompt
 
 @spaces.GPU(duration=120)
 def translate_text(text, source_lang, target_lang):
     if not text.strip():
         return "请输入要翻译的文本"
     try:
+        prompt = prompting(text, source_lang, target_lang)
+        print(prompt)
         input_tokens = (
-            tokenizer(text.strip(), return_tensors="pt")
+            tokenizer(prompt, return_tensors="pt")
             .input_ids[0]
             .cpu()
             .numpy()
@@ -74,7 +77,7 @@ def translate_text(text, source_lang, target_lang):
             num_return_sequences=1,
         )
         full_output = tokenizer.decode(translated_chunk[0], skip_special_tokens=True)
-        full_output = full_output.replace(text.strip(),"")
+        full_output = full_output.replace(prompt.strip(),"")
         yield full_output
     except Exception as e:
         yield f"翻译出错: {str(e)}"