Spaces:
Build error
Build error
Wisdom Chen
commited on
Update model.py
Browse files
model.py
CHANGED
@@ -76,7 +76,7 @@ def initialize_models() -> bool:
|
|
76 |
)
|
77 |
|
78 |
# Get token from Streamlit secrets
|
79 |
-
hf_token = st.secrets
|
80 |
if not hf_token:
|
81 |
raise ValueError("HUGGINGFACE_TOKEN not found in Streamlit secrets")
|
82 |
|
@@ -84,8 +84,7 @@ def initialize_models() -> bool:
|
|
84 |
llm_tokenizer = AutoTokenizer.from_pretrained(
|
85 |
model_name,
|
86 |
token=hf_token,
|
87 |
-
trust_remote_code=True
|
88 |
-
use_fast=False
|
89 |
)
|
90 |
llm_tokenizer.pad_token = llm_tokenizer.eos_token
|
91 |
|
|
|
76 |
)
|
77 |
|
78 |
# Get token from Streamlit secrets
|
79 |
+
hf_token = st.secrets["HUGGINGFACE_TOKEN"]
|
80 |
if not hf_token:
|
81 |
raise ValueError("HUGGINGFACE_TOKEN not found in Streamlit secrets")
|
82 |
|
|
|
84 |
llm_tokenizer = AutoTokenizer.from_pretrained(
|
85 |
model_name,
|
86 |
token=hf_token,
|
87 |
+
trust_remote_code=True
|
|
|
88 |
)
|
89 |
llm_tokenizer.pad_token = llm_tokenizer.eos_token
|
90 |
|