add mean surprisals as output
Browse files
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 |
-
|
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 |
)
|