OLLAMA_URL = "http://127.0.0.1:11434/api/generate"
MODEL = "qwen2.5:7b"
RETRY = 3
SLEEP = 1
# ===== 调用本地大模型 =====
def translate(text, mode="movie"):
if not text:
return None
if mode == "movie":
prompt = f"把下面的英文句子描述翻译成简体中文,保持简洁自然。记住我的要求,前面的部分不要重复, 只给出后面的翻译 :\n{text}"
print(prompt)
for i in range(RETRY):
try:
response =
requests.post(OLLAMA_URL, json={
"model": MODEL,
"prompt": prompt,
"stream": False
}, timeout=120)
result = response.json()
return result.get("response", "").strip()
except Exception as e:
print(f"翻译失败,第{i+1}次重试:", e)
time.sleep(2)
return None
我是用本地模型,效果不行,有的好,有的会有思考过程的结果