ksuzuki01 commited on
Commit
157400d
1 Parent(s): 1abc3e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -2
app.py CHANGED
@@ -42,7 +42,7 @@ def generate(input_text):
42
  output = tokenizer.decode(output_ids.tolist()[0][token_ids.size(1) :], skip_special_tokens=True)
43
  return postprocess(output)
44
 
45
- description: str = """本デモサイトでは個人情報自動マスキングモデル「japanese-gpt-1b-PII-masking」の性能を体験いただけます。
46
  以下の`input`に、個人情報をマスキングしたいテキストデータを入力し「Submit」ボタンを押すと、数秒で処理が完了し`output`に個人情報がマスキングされたテキストデータが表示されます。
47
  """
48
  article: str = (
@@ -59,13 +59,25 @@ article: str = (
59
  | \<company\> | 会社名 |
60
  """
61
  )
 
 
 
 
 
 
62
  iface = gr.Interface(
63
  fn=generate,
64
  inputs=gr.Textbox(label = "input"),
65
- outputs="text",
66
  title = "Japanese GPT 1B PII Masking DEMO",
67
  description=description,
68
  article=article,
 
 
 
 
 
 
69
  )
70
 
71
  iface.launch()
 
42
  output = tokenizer.decode(output_ids.tolist()[0][token_ids.size(1) :], skip_special_tokens=True)
43
  return postprocess(output)
44
 
45
+ description: str = """本デモサイトでは個人情報自動マスキングモデル「japanese-gpt-1b-PII-masking」を体験できます。
46
  以下の`input`に、個人情報をマスキングしたいテキストデータを入力し「Submit」ボタンを押すと、数秒で処理が完了し`output`に個人情報がマスキングされたテキストデータが表示されます。
47
  """
48
  article: str = (
 
59
  | \<company\> | 会社名 |
60
  """
61
  )
62
+ example_input1 = """オペレーター: yabe13@example.co.jpですね。ご確認ありがとうございます。お住まいは東京都江戸川区西瑞江3-1-7、郵便番号は168-5329でよろしいでしょうか?"""
63
+ example_input2 = """オペレーター:ありがとうございます。電話番号を03-2788-7631、住所を東京都台東区浅草橋4-3-8プラネ111とお聞きしました。郵便番号もお持ちでしょうか?
64
+ 中島純治様:あ、郵便番号は、137-6077です。
65
+ オペレーター:137-6077、了解しました。それでは、中島様のご質問について伺ってもよろしいでしょうか?"""
66
+ example_input3 = """オペレーターA:「ありがとうございます、東尾亮介さま、生年月日は2013年8月1日ということですね。お問い合わせの内容について教えていただけますか?」"""
67
+
68
  iface = gr.Interface(
69
  fn=generate,
70
  inputs=gr.Textbox(label = "input"),
71
+ outputs=gr.Textbox(label = "output"),
72
  title = "Japanese GPT 1B PII Masking DEMO",
73
  description=description,
74
  article=article,
75
+ examples = [
76
+ [example_input1],
77
+ [example_input2],
78
+ [example_input3]
79
+ ],
80
+ cache_examples=True
81
  )
82
 
83
  iface.launch()