Update app.py
Browse files
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 |
+
|