strangekitten commited on
Commit
9b1dcbc
1 Parent(s): 3793f45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ abstract = st.text_area("Enter the abstract of the article", height=350)
16
  top_k = st.slider('How many topics from top to show?', 1, 10, 3)
17
  text = title + " " + abstract
18
 
19
- @st.cache(suppress_st_warning=False)
20
  def get_model_tokenizer_linear():
21
  tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-cased")
22
  model = DistilBertModel.from_pretrained("distilbert-base-cased")
 
16
  top_k = st.slider('How many topics from top to show?', 1, 10, 3)
17
  text = title + " " + abstract
18
 
19
+ @st.cache(allow_output_mutation=True)
20
  def get_model_tokenizer_linear():
21
  tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-cased")
22
  model = DistilBertModel.from_pretrained("distilbert-base-cased")