Exched commited on
Commit
92f5a0c
1 Parent(s): a33cc3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ hf_token = os.getenv("HF_TOKEN")
8
  # Model name
9
  model_name = "meta-llama/Llama-2-7b-chat-hf"
10
 
11
- # Load the model and tokenizer with the token
12
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
13
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_token)
14
 
 
8
  # Model name
9
  model_name = "meta-llama/Llama-2-7b-chat-hf"
10
 
11
+ # Load model and tokenizer
12
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
13
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_token)
14