feat(新功能):更换翻译模型,关闭语种判断逻辑 fix(修复bug): refactor(重构): test(增加测试):

This commit is contained in:
zhh
2025-09-26 14:46:03 +08:00
parent 95647be610
commit b8cf3d25b4

View File

@@ -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)