Ayemos commited on
Commit
bf4b865
1 Parent(s): 837a9a4

add mean surprisals as output

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -74,8 +74,7 @@ def main(input_text: str) -> Tuple[float, str]:
74
  input_text, normalize_surprisals=True
75
  )
76
  highlighted_text = create_highlighted_text(tokens2surprisal)
77
- # return mean_surprisal, highlighted_text # not needed for now
78
- return highlighted_text
79
 
80
 
81
  if __name__ == "__main__":
@@ -87,6 +86,7 @@ if __name__ == "__main__":
87
  lines=5, label="テキスト", placeholder="ここにテキストを入力してください。"
88
  ),
89
  outputs=[
 
90
  gr.outputs.HTML(label="トークン毎サプライザル"),
91
  ],
92
  )
 
74
  input_text, normalize_surprisals=True
75
  )
76
  highlighted_text = create_highlighted_text(tokens2surprisal)
77
+ return round(mean_surprisal, 2), highlighted_text
 
78
 
79
 
80
  if __name__ == "__main__":
 
86
  lines=5, label="テキスト", placeholder="ここにテキストを入力してください。"
87
  ),
88
  outputs=[
89
+ gr.Number(label="文全体の読みにくさ(サプライザル)"),
90
  gr.outputs.HTML(label="トークン毎サプライザル"),
91
  ],
92
  )