vinni1484 commited on
Commit
f607c51
·
1 Parent(s): 6d46237

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -1,7 +1,12 @@
1
  import gradio as gr
2
  from keybert import KeyBERT
3
 
4
- model = KeyBERT("sentence-transformers/xlm-r-distilroberta-base-paraphrase-v1")
 
 
 
 
 
5
 
6
  def keywords(text):
7
  keywords = model.extract_keywords(text,top_n=10)
 
1
  import gradio as gr
2
  from keybert import KeyBERT
3
 
4
+ @st.cache(allow_output_mutation=True, suppress_st_warning=True, show_spinner=True)
5
+ def load_model():
6
+ model = KeyBERT("sentence-transformers/xlm-r-distilroberta-base-paraphrase-v1")
7
+ return model
8
+
9
+ model = load_model()
10
 
11
  def keywords(text):
12
  keywords = model.extract_keywords(text,top_n=10)