_

main.py

@@ -25,7 +25,7 @@ device = ("cuda"
 logging.info(C("[INFO] "f"device = {device}"))
 
 # ------------------ INITITALIZE ------------------- #
-@getattr(st, "cache_resource", st.cache)
+@st.cache
 def model_init():
     tokenizer = AutoTokenizer.from_pretrained(
         "ckip-joint/bloom-1b1-zh")