From b8cf3d25b4455ad317bd6dd47ea171f14248ee51 Mon Sep 17 00:00:00 2001 From: zhh Date: Fri, 26 Sep 2025 14:46:03 +0800 Subject: [PATCH] =?UTF-8?q?feat=EF=BC=88=E6=96=B0=E5=8A=9F=E8=83=BD?= =?UTF-8?q?=EF=BC=89:=E6=9B=B4=E6=8D=A2=E7=BF=BB=E8=AF=91=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=EF=BC=8C=E5=85=B3=E9=97=AD=E8=AF=AD=E7=A7=8D=E5=88=A4?= =?UTF-8?q?=E6=96=AD=E9=80=BB=E8=BE=91=20=20fix=EF=BC=88=E4=BF=AE=E5=A4=8D?= =?UTF-8?q?bug=EF=BC=89:=20refactor=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:?= =?UTF-8?q?=20test(=E5=A2=9E=E5=8A=A0=E6=B5=8B=E8=AF=95):?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../prompt_generation/chatgpt_for_translation.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/app/service/prompt_generation/chatgpt_for_translation.py b/app/service/prompt_generation/chatgpt_for_translation.py index 4c50d0b..7933952 100644 --- a/app/service/prompt_generation/chatgpt_for_translation.py +++ b/app/service/prompt_generation/chatgpt_for_translation.py @@ -90,20 +90,20 @@ def get_response(messages): def get_translation_from_llama3(text): start_time = time.time() - url = "http://10.1.1.240:11434/api/generate" - # url = "http://10.1.1.240:1143/api/generate" + # url = "http://10.1.1.240:11434/api/generate" + url = "http://10.1.1.243:11434/api/generate" # prompt = f"System: {prefix_for_llama}\nUser:[{text}]" # 先获取用户输入文本的语言 - language = get_language(text) + # language = get_language(text) - if 'English' in language: - return text + # if 'English' in language: + # return text # 创建请求的负载 translator是自定义的翻译模型 payload = { - "model": "translator", + "model": "zcr_gemma3_translator:4b", "prompt": f"[{text}]", "stream": False } @@ -178,7 +178,7 @@ def get_prompt_from_image(image_path, text): def main(): """Main function""" - text = get_translation_from_llama3("[火焰]") + text = get_translation_from_llama3("[123]") print(text)