当前位置:   article > 正文

Python离线翻译

python离线翻译

Python源码

  1. # 离线翻译服务代码
  2. import os
  3. from flask import Flask, request
  4. from gevent import pywsgi
  5. from transformers import pipeline, AutoModelWithLMHead, AutoTokenizer
  6. import warnings, requests
  7. warnings.filterwarnings('ignore')
  8. try:
  9. print('正在加载【汉语-英语】翻译模型... ...')
  10. model = AutoModelWithLMHead.from_pretrained('Helsinki-NLP/opus-mt-zh-en')
  11. tokenizer = AutoTokenizer.from_pretrained('Helsinki-NLP/opus-mt-zh-en')
  12. translation = pipeline('translation_zh_to_en', model=model, tokenizer=tokenizer)
  13. print('正在加载【英语-汉语】翻译模型... ...')
  14. model_en2zh = AutoModelWithLMHead.from_pretrained('Helsinki-NLP/opus-mt-en-zh')
  15. tokenizer_en2zh = AutoTokenizer.from_pretrained('Helsinki-NLP/opus-mt-en-zh')
  16. translation_en2zh = pipeline('translation_en_to_zh', model=model_en2zh, tokenizer=tokenizer_en2zh)
  17. app = Flask(__
本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号