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()