Nikendolo commited on
Commit
da31c61
1 Parent(s): 762d381

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,7 +72,7 @@ def predict(text, model, human_readable=True):
72
  return answer
73
 
74
 
75
- @st.cache(suppress_st_warning=True)
76
  def load_model_and_tokenizer():
77
  return (torch.load(model_path, map_location=torch.device(device)),
78
  DistilBertTokenizer.from_pretrained(vocab_path))
 
72
  return answer
73
 
74
 
75
+ @st.cache(show_spinner=False, allow_output_mutation=True)
76
  def load_model_and_tokenizer():
77
  return (torch.load(model_path, map_location=torch.device(device)),
78
  DistilBertTokenizer.from_pretrained(vocab_path))