mbahrami commited on
Commit
89ab94a
·
1 Parent(s): ec37d6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def get_model(model):
15
  def loading_models(model='roberta-base'):
16
  return get_model(model), SentenceTransformer('all-MiniLM-L6-v2')
17
 
18
- @st.cache(allow_output_mutation=True)
19
  def main(nlp, semantic_model, data_load_state):
20
  data_load_state.text('Inference from model...')
21
  result = nlp(text+' '+nlp.tokenizer.mask_token)
 
15
  def loading_models(model='roberta-base'):
16
  return get_model(model), SentenceTransformer('all-MiniLM-L6-v2')
17
 
18
+ #@st.cache(allow_output_mutation=True)
19
  def main(nlp, semantic_model, data_load_state):
20
  data_load_state.text('Inference from model...')
21
  result = nlp(text+' '+nlp.tokenizer.mask_token)