赞
踩
-
- # 离线翻译服务代码
- 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(__

Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。