InitialCommit
This commit is contained in:
commit
86dc185f1e
|
|
@ -0,0 +1,4 @@
|
|||
Lib/
|
||||
Scripts/
|
||||
Include/
|
||||
pyvenv.cfg
|
||||
|
|
@ -0,0 +1,21 @@
|
|||
from deep_translator import (GoogleTranslator, PapagoTranslator, BaiduTranslator)
|
||||
|
||||
def google_translate(text, target_lang='ko'):
|
||||
translated = GoogleTranslator(source='auto', target=target_lang).translate(text)
|
||||
return f"Google Translate: {translated}"
|
||||
|
||||
def papago_translate(text, target_lang='ko'):
|
||||
translated = PapagoTranslator(client_id='YOUR_CLIENT_ID', secret_key='YOUR_SECRET_KEY', source='auto', target=target_lang).translate(text)
|
||||
return f"Papago Translate: {translated}"
|
||||
|
||||
def baidu_translate(text, target_lang='zh'):
|
||||
translated = BaiduTranslator(appid='YOUR_APP_ID', appkey='YOUR_APP_KEY', source='auto', target=target_lang).translate(text)
|
||||
return f"Baidu Translate: {translated}"
|
||||
|
||||
# Example usage
|
||||
text = "百度AI开放平台 \n 全球领先的人工智能服务平台"
|
||||
|
||||
print(google_translate(text))
|
||||
print(papago_translate(text))
|
||||
print(baidu_translate(text))
|
||||
|
||||
|
|
@ -0,0 +1,73 @@
|
|||
from transformers import MarianMTModel, MarianTokenizer, MBartForConditionalGeneration, MBart50TokenizerFast
|
||||
import requests
|
||||
import hashlib
|
||||
import random
|
||||
import urllib.request
|
||||
import json
|
||||
|
||||
# Baidu API 설정
|
||||
BAIDU_APP_ID = 'YOUR_APP_ID'
|
||||
BAIDU_SECRET_KEY = 'YOUR_SECRET_KEY'
|
||||
|
||||
# Papago API 설정
|
||||
PAPAGO_CLIENT_ID = 'YOUR_CLIENT_ID'
|
||||
PAPAGO_CLIENT_SECRET = 'YOUR_CLIENT_SECRET'
|
||||
|
||||
def marian_translate_zh_ko(text):
|
||||
model_name = 'Helsinki-NLP/opus-mt-zh-ko'
|
||||
tokenizer = MarianTokenizer.from_pretrained(model_name)
|
||||
model = MarianMTModel.from_pretrained(model_name)
|
||||
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
|
||||
translated = model.generate(**inputs)
|
||||
translated_text = tokenizer.batch_decode(translated, skip_special_tokens=True)[0]
|
||||
return f"MarianMT Translate (zh-ko): {translated_text}"
|
||||
|
||||
def mbart_translate_zh_ko(text):
|
||||
model_name = 'facebook/mbart-large-50-many-to-many-mmt'
|
||||
tokenizer = MBart50TokenizerFast.from_pretrained(model_name)
|
||||
model = MBartForConditionalGeneration.from_pretrained(model_name)
|
||||
tokenizer.src_lang = 'zh_CN'
|
||||
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
|
||||
translated = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id['ko_KR'])
|
||||
translated_text = tokenizer.batch_decode(translated, skip_special_tokens=True)[0]
|
||||
return f"mBART Translate (zh-ko): {translated_text}"
|
||||
|
||||
def baidu_translate_zh_ko(text):
|
||||
url = "https://fanyi-api.baidu.com/api/trans/vip/translate"
|
||||
salt = random.randint(32768, 65536)
|
||||
sign = hashlib.md5((BAIDU_APP_ID + text + str(salt) + BAIDU_SECRET_KEY).encode('utf-8')).hexdigest()
|
||||
params = {
|
||||
'q': text,
|
||||
'from': 'zh',
|
||||
'to': 'kor',
|
||||
'appid': BAIDU_APP_ID,
|
||||
'salt': salt,
|
||||
'sign': sign
|
||||
}
|
||||
response = requests.get(url, params=params)
|
||||
result = response.json()
|
||||
return f"Baidu Translate (zh-ko): {result['trans_result'][0]['dst']}"
|
||||
|
||||
def papago_translate_zh_ko(text):
|
||||
encText = urllib.parse.quote(text)
|
||||
data = "source=zh-CN&target=ko&text=" + encText
|
||||
url = "https://openapi.naver.com/v1/papago/n2mt"
|
||||
request = urllib.request.Request(url)
|
||||
request.add_header("X-Naver-Client-Id", PAPAGO_CLIENT_ID)
|
||||
request.add_header("X-Naver-Client-Secret", PAPAGO_CLIENT_SECRET)
|
||||
response = urllib.request.urlopen(request, data=data.encode("utf-8"))
|
||||
rescode = response.getcode()
|
||||
if(rescode==200):
|
||||
response_body = response.read()
|
||||
result = json.loads(response_body.decode('utf-8'))
|
||||
return f"Papago Translate (zh-ko): {result['message']['result']['translatedText']}"
|
||||
else:
|
||||
return "Error: Papago translation failed."
|
||||
|
||||
# Example usage
|
||||
text = "你好"
|
||||
|
||||
print(marian_translate_zh_ko(text))
|
||||
print(mbart_translate_zh_ko(text))
|
||||
print(baidu_translate_zh_ko(text))
|
||||
print(papago_translate_zh_ko(text))
|
||||
Loading…
Reference in New Issue