Python离线翻译

zow

已于 2023-04-20 15:20:15 修改

阅读量3k

点赞数 4

分类专栏：小方法文章标签： python flask 开发语言

于 2022-10-07 10:48:44 首次发布

本文链接：https://blog.csdn.net/lenkty/article/details/127121655

版权

这篇博客介绍了如何利用Python进行离线翻译，关键代码中涉及到了huggingface-Helsinki-NLP翻译模型的选择与调用。在启动服务时，可能会遇到需要安装额外插件和下载模型的情况，Postman被用来进行功能测试。

摘要由CSDN通过智能技术生成

Python源码：


# 离线翻译服务代码
import os
from flask import Flask, request
from gevent import pywsgi
from transformers import pipeline, AutoModelWithLMHead, AutoTokenizer
import warnings, requests
warnings.filterwarnings('ignore')

try:    
    print('正在加载【汉语-英语】翻译模型... ...')
    model = AutoModelWithLMHead.from_pretrained('Helsinki-NLP/opus-mt-zh-en')
    tokenizer = AutoTokenizer.from_pretrained('Helsinki-NLP/opus-mt-zh-en')
    translation = pipeline('translation_zh_to_en', model=model, tokenizer=tokenizer)
    print('正在加载【英语-汉语】翻译模型... ...')
    model_en2zh = AutoModelWithLMHead.from_pretrained('Helsinki-NLP/opus-mt-en-zh')
    tokenizer_en2zh = AutoTokenizer.from_pretrained('Helsinki-NLP/opus-mt-en-zh')
    translation_en2zh = pipeline('translation_en_to_zh', model=model_en2zh, tokenizer=tokenizer_en2zh)
    
    app = Flask(__