document-translator / src /mtuoc_aina_translator.py
mjuvilla's picture
Updated readme and added salamandraTA7b translator class
bc3b289
raw
history blame contribute delete
537 Bytes
import requests
import json
class MTUOCAinaTranslator:
def __init__(self, ip: str, port: str):
self.ip = ip
self.port = port
def translate(self, text, source_lang=None, target_lang=None):
myobj = {
'id': '1',
'src': text,
}
url = f'http://{self.ip}:{self.port}/translate'
#url = 'http://' + self.ip + ':' + self.port + '/translate'
x = requests.post(url, json=myobj)
json_response = json.loads(x.text)
return json_response['tgt']