sidcww commited on
Commit
ffd4917
·
verified ·
1 Parent(s): 8a18cb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -1,6 +1,3 @@
1
- # -*- coding: utf-8 -*-
2
- """keyword_extraction"""
3
-
4
  import requests
5
  import jieba
6
  from keybert import KeyBERT
@@ -61,23 +58,21 @@ doc = st.text_area("請輸入文章:")
61
 
62
  if st.button("提取關鍵詞"):
63
  if doc:
64
- # 提取關鍵詞
65
  keywords = extract_keywords(doc)
66
  st.write("關鍵詞提取結果:")
67
  for keyword in keywords:
68
  st.write(f"{keyword[0]}: {keyword[1]:.4f}")
69
 
70
- # 顯示關鍵詞圖
71
  plot_keywords(keywords, "關鍵詞提取結果")
72
 
73
- # 使用多語言模型進行關鍵詞提取
74
  kw_model_multilingual = KeyBERT(model='distiluse-base-multilingual-cased-v1')
75
  keywords_multilingual = kw_model_multilingual.extract_keywords(doc, vectorizer=vectorizer)
76
  st.write("多語言模型關鍵詞提取結果:")
77
  for keyword in keywords_multilingual:
78
  st.write(f"{keyword[0]}: {keyword[1]:.4f}")
79
 
80
- # 顯示多語言模型關鍵詞圖
81
  plot_keywords(keywords_multilingual, "多語言模型關鍵詞提取結果")
82
  else:
83
  st.write("請輸入文章內容以進行關鍵詞提取。")
 
 
 
 
 
1
  import requests
2
  import jieba
3
  from keybert import KeyBERT
 
58
 
59
  if st.button("提取關鍵詞"):
60
  if doc:
 
61
  keywords = extract_keywords(doc)
62
  st.write("關鍵詞提取結果:")
63
  for keyword in keywords:
64
  st.write(f"{keyword[0]}: {keyword[1]:.4f}")
65
 
 
66
  plot_keywords(keywords, "關鍵詞提取結果")
67
 
68
+ # 使用另一個模型進行關鍵詞提取
69
  kw_model_multilingual = KeyBERT(model='distiluse-base-multilingual-cased-v1')
70
  keywords_multilingual = kw_model_multilingual.extract_keywords(doc, vectorizer=vectorizer)
71
  st.write("多語言模型關鍵詞提取結果:")
72
  for keyword in keywords_multilingual:
73
  st.write(f"{keyword[0]}: {keyword[1]:.4f}")
74
 
 
75
  plot_keywords(keywords_multilingual, "多語言模型關鍵詞提取結果")
76
  else:
77
  st.write("請輸入文章內容以進行關鍵詞提取。")
78
+