Hugging Face's logo

Spaces:
jeffeux
/
ckip-joint_bloom_streamlit
Runtime error

App Files Community
jeffeux commited on
Commit
e0fbe2a
·
1 Parent(s): 8446129

_

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -25,7 +25,7 @@ device = ("cuda"
25
  logging.info(C("[INFO] "f"device = {device}"))
26
 
27
  # ------------------ INITITALIZE ------------------- #
28
- @getattr(st, "cache_resource", st.cache)
29
  def model_init():
30
  tokenizer = AutoTokenizer.from_pretrained(
31
  "ckip-joint/bloom-1b1-zh")
 
25
  logging.info(C("[INFO] "f"device = {device}"))
26
 
27
  # ------------------ INITITALIZE ------------------- #
28
+ @st.cache
29
  def model_init():
30
  tokenizer = AutoTokenizer.from_pretrained(
31
  "ckip-joint/bloom-1b1-zh")