Spaces:
Runtime error
Runtime error
File size: 680 Bytes
a284249 5dd2a47 a284249 5dd2a47 474b498 5dd2a47 |
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.")
|