Spaces:
Runtime error
Runtime error
File size: 439 Bytes
ef0970f 086e787 f607c51 9a41298 ef0970f 6d46237 ef0970f 086e787 6d46237 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import gradio as gr
from keybert import KeyBERT
@st.cache(allow_output_mutation=True, suppress_st_warning=True, show_spinner=True)
def load_model():
model = KeyBERT("sentence-transformers/xlm-r-distilroberta-base-paraphrase-v1")
return model
model = load_model()
def keywords(text):
keywords = model.extract_keywords(text,top_n=10)
return keywords
gr.Interface(keywords, "text", "text",title="Keyword Extractor").launch() |