cache fix

src/main.py

@@ -18,7 +18,7 @@ def label_dictionary(model_name):
         return twitter_roberta
     return lambda x: x
 
-@st.cache_resource
+@st.cache(allow_output_mutation=True)
 def load_model(model_name):
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained(model_name)