Update app.py
Browse files
app.py
CHANGED
@@ -72,7 +72,7 @@ def predict(text, model, human_readable=True):
|
|
72 |
return answer
|
73 |
|
74 |
|
75 |
-
@st.cache(
|
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))
|