Update app.py
Browse files
app.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
-
import transformers
|
| 3 |
|
| 4 |
encoder_model_name = "cl-tohoku/bert-base-japanese-v2"
|
| 5 |
decoder_model_name = "skt/kogpt2-base-v2"
|
|
@@ -7,6 +7,7 @@ src_tokenizer = transformers.BertJapaneseTokenizer.from_pretrained(encoder_model
|
|
| 7 |
trg_tokenizer = transformers.PreTrainedTokenizerFast.from_pretrained(decoder_model_name)
|
| 8 |
model = transformers.EncoderDecoderModel.from_pretrained("sappho192/ffxiv-ja-ko-translator")
|
| 9 |
|
|
|
|
| 10 |
def translate(text_src):
|
| 11 |
embeddings = src_tokenizer(text_src, return_attention_mask=False, return_token_type_ids=False, return_tensors='pt')
|
| 12 |
embeddings = {k: v for k, v in embeddings.items()}
|
|
@@ -14,9 +15,11 @@ def translate(text_src):
|
|
| 14 |
text_trg = trg_tokenizer.decode(output.cpu())
|
| 15 |
return text_trg
|
| 16 |
|
|
|
|
| 17 |
def endpoint(sentence):
|
| 18 |
return translate(sentence)
|
| 19 |
|
|
|
|
| 20 |
# demo = gr.Interface(fn=endpoint, inputs="text", outputs="text")
|
| 21 |
with gr.Blocks() as demo:
|
| 22 |
input = gr.Textbox(label="Sentence")
|
|
@@ -42,4 +45,4 @@ with gr.Blocks() as demo:
|
|
| 42 |
)
|
| 43 |
|
| 44 |
if __name__ == "__main__":
|
| 45 |
-
demo.launch()
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
+
import transformers
|
| 3 |
|
| 4 |
encoder_model_name = "cl-tohoku/bert-base-japanese-v2"
|
| 5 |
decoder_model_name = "skt/kogpt2-base-v2"
|
|
|
|
| 7 |
trg_tokenizer = transformers.PreTrainedTokenizerFast.from_pretrained(decoder_model_name)
|
| 8 |
model = transformers.EncoderDecoderModel.from_pretrained("sappho192/ffxiv-ja-ko-translator")
|
| 9 |
|
| 10 |
+
|
| 11 |
def translate(text_src):
|
| 12 |
embeddings = src_tokenizer(text_src, return_attention_mask=False, return_token_type_ids=False, return_tensors='pt')
|
| 13 |
embeddings = {k: v for k, v in embeddings.items()}
|
|
|
|
| 15 |
text_trg = trg_tokenizer.decode(output.cpu())
|
| 16 |
return text_trg
|
| 17 |
|
| 18 |
+
|
| 19 |
def endpoint(sentence):
|
| 20 |
return translate(sentence)
|
| 21 |
|
| 22 |
+
|
| 23 |
# demo = gr.Interface(fn=endpoint, inputs="text", outputs="text")
|
| 24 |
with gr.Blocks() as demo:
|
| 25 |
input = gr.Textbox(label="Sentence")
|
|
|
|
| 45 |
)
|
| 46 |
|
| 47 |
if __name__ == "__main__":
|
| 48 |
+
demo.launch()
|