Helsinki-NLP / OPUS-MT-train

Training open neural machine translation models
MIT License
323 stars 40 forks source link

How to translate from english to Japan? #82

Closed izzahzawave closed 1 year ago

izzahzawave commented 1 year ago

I use flask to do the language translation from en-jp, jp to en. But the issue is I got some error when translating from English to Japan, it says that the file is not found. Here is the code that I used to translate from Japan to English. Any idea on how to do the opposite ?

from translate import Translator from . import languageTranslator import flask from flask import Flask, jsonify, request from transformers import MarianMTModel, MarianTokenizer from typing import Sequence

src = 'ja' trg = 'en'
modelName = f'Helsinki-NLP/opus-mt-{src}-{trg}'

model = MarianMTModel.from_pretrained(modelName) tokenizer = MarianTokenizer.from_pretrained(modelName)

def predict(sampleText): batch = tokenizer(sampleText, return_tensors="pt", padding = True) gen = model.generate(**batch) return tokenizer.batch_decode(gen, skip_special_tokens=True)[0]

@languageTranslator.route("/languageTranslate/", methods=["POST"]) def translator(text): try: translationText = predict(text)

    return flask.jsonify({
        'result': translationText
        })
except Exception as e:
    return str(e)