show mean surprisal

app.py

@@ -112,8 +112,12 @@ def highlight_token(token: str, score: float):
     )
 
 
-def create_highlighted_text(label: str, tokens2scores: List[Tuple[str, float]]):
-    highlighted_text: str = "<h2><b>" + label + "</b></h2>"
+def create_highlighted_text(
+    label: str, tokens2scores: List[Tuple[str, float]], mean_surprisal: float
+):
+    highlighted_text: str = (
+        "<h2><b>" + label + f"</b>（サプライザル平均値: {mean_surprisal}）</h2>"
+    )
     for token, score in tokens2scores:
         highlighted_text += highlight_token(token, score)
     return highlighted_text
@@ -161,7 +165,9 @@ def main(input_text: str) -> Tuple[str, str, str]:
     offsets = calc_offsets(sudachi_tokenize(input_text))
     tokens2surprisal = aggregate_surprisals_by_offset(char2surprisal, offsets)
     tokens2surprisal = normalize_surprisals(tokens2surprisal)
-    highlighted_text = create_highlighted_text("学習後モデル", tokens2surprisal)
+    highlighted_text = create_highlighted_text(
+        "学習後モデル", tokens2surprisal, mean_surprisal
+    )
 
     (
         baseline_mean_surprisal,
@@ -172,7 +178,7 @@ def main(input_text: str) -> Tuple[str, str, str]:
     )
     baseline_tokens2surprisal = normalize_surprisals(baseline_tokens2surprisal)
     baseline_highlighted_text = create_highlighted_text(
-        "学習前モデル", baseline_tokens2surprisal
+        "学習前モデル", baseline_tokens2surprisal, baseline_mean_surprisal
     )
 
     diff_tokens2surprisal = calculate_surprisal_diff(
@@ -189,7 +195,7 @@ def main(input_text: str) -> Tuple[str, str, str]:
 if __name__ == "__main__":
     demo = gr.Interface(
         fn=main,
-        title="文章の読みやすさを自動評価する言語処理システム",
+        title="読みにくい箇所を検出するAI（デモ）",
         description="テキストを入力すると、読みにくさに応じてハイライトされて出力されます。",
         show_label=True,
         inputs=gr.Textbox(