Update app.py

app.py

@@ -25,6 +25,7 @@ def predict_sentence(model, tokenizer, sentence):
     title = convert_label_to_title(predicted_class)
     return title, probabilities.numpy(), highest_probability
 
+@st.cache(allow_output_mutation=True)
 def load_model(checkpoint, num_class):
   model = TFAutoModelForSequenceClassification.from_pretrained(checkpoint, num_labels=num_class)
   tokenizer = AutoTokenizer.from_pretrained(checkpoint)