rzaytsev commited on
Commit
658f952
1 Parent(s): 959ca81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,11 +5,11 @@ import torch
5
 
6
  model_name = 'distilbert-base-cased'
7
 
8
- @st.cache(suppress_st_warning=True)
9
  def load_model():
10
  return AutoTokenizer.from_pretrained(model_name), AutoModelForSequenceClassification.from_pretrained('./')
11
 
12
- tokenizer, model = load_model()
 
13
 
14
  title = st.text_area('Title')
15
  abstract = st.text_area('Abstract')
 
5
 
6
  model_name = 'distilbert-base-cased'
7
 
 
8
  def load_model():
9
  return AutoTokenizer.from_pretrained(model_name), AutoModelForSequenceClassification.from_pretrained('./')
10
 
11
+ if 'tokenizer' not in globals():
12
+ tokenizer, model = load_model()
13
 
14
  title = st.text_area('Title')
15
  abstract = st.text_area('Abstract')