Hugging Face's logo

Spaces:
MSey
/
CaGBERT
Sleeping

App Files Community
MSey90 commited on
Commit
bcc7480
·
unverified ·
1 Parent(s): 9452458

updates cahce decorator

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import streamlit as st
5
  # Access the secret token
6
  hf_token = st.secrets["default"]["hf_token"]
7
 
8
- @st.cache(allow_output_mutation=True)
9
  def load_model():
10
  return pipeline("text-generation", model="MSey/tiny_CaLL_r10_O1_f10_LT_c1022", use_auth_token =hf_token)
11
 
 
5
  # Access the secret token
6
  hf_token = st.secrets["default"]["hf_token"]
7
 
8
+ @st.cache_resource
9
  def load_model():
10
  return pipeline("text-generation", model="MSey/tiny_CaLL_r10_O1_f10_LT_c1022", use_auth_token =hf_token)
11