Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from keybert import KeyBERT
+import jieba
 
 # 初始化 KeyBERT 模型
 kw_model = KeyBERT()
@@ -13,7 +14,10 @@ text = st.text_area("請貼上文字並按下按鈕以抓取關鍵字", height=2
 # 按鈕來觸發關鍵字抓取
 if st.button("抓取關鍵字"):
     if text:
-        keywords = kw_model.extract_keywords(text, stop_words='english')
+        # 使用 jieba 對輸入的中文文本進行分詞
+        words = " ".join(jieba.cut(text))
+        # 使用 KeyBERT 抓取關鍵字
+        keywords = kw_model.extract_keywords(words, stop_words='english')
         st.write("抓取到的關鍵字及相關性分數:")
         for keyword, relevance in keywords:
             st.write(f"關鍵字: {keyword}, 相關性分數: {relevance:.4f}")