lmarty commited on
Commit
e168c4b
·
verified ·
1 Parent(s): 64bcb68

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -175,7 +175,11 @@ def show_mitigation(bias_type):
175
  def setup_models():
176
  hf_token = os.getenv("fairLLM")
177
 
178
- tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct")
 
 
 
 
179
  #set pad_token_id to eos_token_id if it's not already defined
180
  if tokenizer.pad_token_id is None:
181
  tokenizer.pad_token_id = tokenizer.eos_token_id
 
175
  def setup_models():
176
  hf_token = os.getenv("fairLLM")
177
 
178
+ tokenizer = AutoTokenizer.from_pretrained(
179
+ "meta-llama/Llama-3.2-1B-Instruct",
180
+ token=hf_token
181
+ )
182
+
183
  #set pad_token_id to eos_token_id if it's not already defined
184
  if tokenizer.pad_token_id is None:
185
  tokenizer.pad_token_id = tokenizer.eos_token_id