Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -71,7 +71,7 @@ def load_llm_model():
|
|
71 |
llm = AutoModelForCausalLM.from_pretrained('PyaeSoneK/LlamaV2LegalFineTuned',
|
72 |
device_map='auto',
|
73 |
torch_dtype=torch.float16,
|
74 |
-
use_auth_token= st.secrets['
|
75 |
)
|
76 |
return llm
|
77 |
|
|
|
71 |
llm = AutoModelForCausalLM.from_pretrained('PyaeSoneK/LlamaV2LegalFineTuned',
|
72 |
device_map='auto',
|
73 |
torch_dtype=torch.float16,
|
74 |
+
use_auth_token= st.secrets['hf_access_token'],
|
75 |
)
|
76 |
return llm
|
77 |
|