Spaces:
Runtime error
Runtime error
File size: 694 Bytes
a284249 5dd2a47 a284249 5dd2a47 e89867e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import streamlit as st
from keybert import KeyBERT
# KeyBERT model
keybert_model = KeyBERT(model='all-MiniLM-L6-v2')
def get_bert_keywords(text, num_keywords):
result = keybert_model.extract_keywords(text, keyphrase_ngram_range=(1, 1), top_n=num_keywords)
keyberts = [keyword for keyword, score in result]
return keyberts
# Streamlit app
st.title("Keyword Extraction with KeyBERT")
text = st.text_area("Enter your text here:", value='', height=200)
# if st.button("Extract Keywords"):
# if text:
# keywords = get_bert_keywords(text, 5)
# st.write("Keywords:")
# st.write(", ".join(keywords))
# else:
# st.error("Please enter some text.")
|