File size: 598 Bytes
247182b
0fdff2d
 
247182b
 
 
0fdff2d
247182b
0fdff2d
 
247182b
 
 
0fdff2d
247182b
 
0fdff2d
247182b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from transformers import AutoModelForSeq2SeqLM,AutoTokenizer
import gradio as grad

mdl_name = "Helsinki-NLP/opus-mt-ko-en"
mdl = AutoModelForSeq2SeqLM.from_pretrained(mdl_name)
my_tkn = AutoTokenizer.from_pretrained(mdl_name)

#opus_translator = pipeline("translation", model=mdl_name)

def translate(text):
    inputs = my_tkn(text, return_tensors="pt")
    trans_output = mdl.generate(**inputs)
    response = my_tkn.decode(trans_output[0], skip_special_tokens=True)

    #response = opus_translator(text)
    return response

grad.Interface(translate, inputs=["text",], outputs="text").launch()