update

app.py

@@ -42,35 +42,32 @@ def main():
 
     st.title("SPLADE 日本語 demo")
 
+    get_embedder()
+
     st.markdown("""
-    ### 入力
-    以下のテキストエリアに解析したいテキストを入力してください。
+    [hotchpotch/japanese-splade-base-v1](https://huggingface.co/hotchpotch/japanese-splade-base-v1)を使って、テキストからSPLADEのスパースベクトルに変換するデモです。
+
     """)
 
-    input_text = st.text_area("テキスト入力", height=200)
+    input_text = st.text_area("テキスト", height=200)
 
-    if st.button("解析開始"):
+    if st.button("変換"):
         if input_text.strip():
-            with st.spinner("解析中..."):
+            with st.spinner("変換中..."):
                 sorted_tokens = get_token_values_sorted(input_text)
 
-            st.success("解析が完了しました。")
-
-            st.markdown("### 結果")
+            total_tokens = len(sorted_tokens)
+            st.markdown(f"### 結果 (トークン数: {total_tokens})")
             if sorted_tokens:
                 formatted_data = [
-                    {"頻度": freq, "単語": word} for freq, word in sorted_tokens
+                    {"スコア": freq, "単語(vocab)": word}
+                    for freq, word in sorted_tokens
                 ]
                 st.table(formatted_data)
             else:
                 st.warning("入力テキストから有効な単語が見つかりませんでした。")
         else:
             st.warning("テキストを入力してください。")
-    else:
-        get_embedder()
-
-    st.markdown("---")
-    st.markdown("© 2024 SPLADE 日本語 demo")
 
 
 if __name__ == "__main__":