fix

st_utils.py

@@ -93,7 +93,7 @@ class CONFIG:
 
 
 # download model with streamlit cache decorator
-@st.cache(persist=False, show_spinner=True)
+@st.cache(persist=False, show_spinner=True, allow_output_mutation=True)
 def download_model():
     if not os.path.exists(r"models/pytorch_model.bin"):
         os.makedirs("./models", exist_ok=True)
@@ -108,7 +108,7 @@ def download_model():
 
 
 # load with streamlit cache decorator
-@st.cache(persist=False, show_spinner=True)
+@st.cache(persist=False, show_spinner=True, allow_output_mutation=True)
 def load_tokenizer_and_model(pretrained_path):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")