diff --git a/app/service/prompt_generation/chatgpt_for_translation.py b/app/service/prompt_generation/chatgpt_for_translation.py index 7933952..4c50d0b 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.243:11434/api/generate" + url = "http://10.1.1.240:11434/api/generate" + # url = "http://10.1.1.240:1143/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": "zcr_gemma3_translator:4b", + "model": "translator", "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("[123]") + text = get_translation_from_llama3("[火焰]") print(text)