Spaces:
Runtime error
Runtime error
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()
|