File size: 709 Bytes
a84af81
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
import gradio as grad

mdl_name = "Helsinki-NLP/opus-mt-en-de"

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()