Spaces:
Runtime error
Runtime error
d.tsimerman
commited on
Commit
•
f5e42ed
1
Parent(s):
c7ca481
good
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ model_name = st.selectbox(
|
|
12 |
)
|
13 |
auth_token = os.environ.get('TOKEN') or True
|
14 |
|
15 |
-
@st.cache(hash_funcs={tokenizers.Tokenizer: lambda tokenizer: hash(tokenizer.
|
16 |
def load_model(model_name: str):
|
17 |
with st.spinner('Loading models...'):
|
18 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)
|
12 |
)
|
13 |
auth_token = os.environ.get('TOKEN') or True
|
14 |
|
15 |
+
@st.cache(hash_funcs={tokenizers.Tokenizer: lambda tokenizer: hash(tokenizer.to_str())})
|
16 |
def load_model(model_name: str):
|
17 |
with st.spinner('Loading models...'):
|
18 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)
|