update column name

app.py

@@ -68,13 +68,15 @@ def get_stats():
 
     # if reading != '<OTHER>'
 
-    df = pd.DataFrame({"heteronym": heteronyms, "accuracy": accuracy, "readings": readings})
+    df = pd.DataFrame(
+        {"heteronym": heteronyms, "accuracy": accuracy, "readings": readings}
+    )
 
     df = df[df["readings"].str.contains("、")]
 
     df["readings"] = df["readings"].str.replace("<OTHER>", "Other")
 
-    df = df.rename(columns={"readings": "readings (test corr./total)"})
+    df = df.rename(columns={"readings": "readings (correct/total)"})
 
     df = df.sort_values("accuracy", ascending=False, ignore_index=True)
 
@@ -113,7 +115,7 @@ def furigana_to_spacy(text_with_furigana):
     }
 
 
-st.title("Yomikata: Disambiguate Japanese Heteronyms with a BERT model")
+st.title("Yomikata: Disambiguate Japanese Heteronyms")
 
 # Input text box
 st.markdown("Input a Japanese sentence:")
@@ -137,10 +139,14 @@ label_colors = {
     reading: colors[i % len(colors)]
     for i, reading in enumerate(set([item["label"] for item in spacy_dict["ents"]]))
 }
-html = spacy.displacy.render(spacy_dict, style="ent", manual=True, options={"colors": label_colors})
+html = spacy.displacy.render(
+    spacy_dict, style="ent", manual=True, options={"colors": label_colors}
+)
 
 if len(spacy_dict["ents"]) > 0:
-    st.markdown("**Yomikata** found and disambiguated the following heteronyms:")
+    st.markdown(
+        "**Yomikata** disambiguated the following words with multiple readings:"
+    )
     st.write(
         f"{add_border(html)}",
         unsafe_allow_html=True,
@@ -184,11 +190,15 @@ if st.button("🎲 Randomize the input sentence"):
 global_accuracy, stats_df = get_stats()
 
 st.subheader(
-    f"{len(stats_df)} heteronyms supported, with a global accuracy of {global_accuracy:.0%}"
+    f"**Yomikata** supports {len(stats_df)} heteronyms, with a global accuracy of {global_accuracy:.0%}!"
 )
 
 st.dataframe(stats_df)
 
+st.subheader(
+    "Check out **Yomikata** on [GitHub](https://github.com/passaglia/yomikata) today!"
+)
+
 # Hide the footer
 hide_streamlit_style = """
             <style>