Hugging Face's logo

Spaces:
GT4SD
/
PatentToolkit
Runtime error

App Files Community
3
thepolymerguy commited on
Commit
d5be6af
·
1 Parent(s): 906763f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -7,9 +7,11 @@ import transformers
7
  from transformers import AutoTokenizer, AutoConfig, LlamaForCausalLM, LlamaTokenizer, GenerationConfig
8
 
9
 
10
- auth_token = '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' #= os.environ.get("TOKEN_FROM_SECRET")
 
11
 
12
- # tokenizer = LlamaTokenizer.from_pretrained("Claimed/capybara", use_auth_token=auth_token)
 
13
 
14
  # model = LlamaForCausalLM.from_pretrained(
15
  # "Claimed/capybara", use_auth_token=auth_token)
 
7
  from transformers import AutoTokenizer, AutoConfig, LlamaForCausalLM, LlamaTokenizer, GenerationConfig
8
 
9
 
10
+ #auth_token = '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' #
11
+ auth_token = os.environ.get("AUTH_TOKEN_SECRET")
12
 
13
+
14
+ tokenizer = LlamaTokenizer.from_pretrained("Claimed/capybara", use_auth_token=auth_token)
15
 
16
  # model = LlamaForCausalLM.from_pretrained(
17
  # "Claimed/capybara", use_auth_token=auth_token)