sappho192 commited on
Commit
4966749
·
1 Parent(s): 4004987

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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()