feat(新功能):更换翻译模型,关闭语种判断逻辑 fix(修复bug): refactor(重构): test(增加测试):
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user