jbdel commited on
Commit
8f1a208
1 Parent(s): c095dce

push with token

Browse files
Files changed (1) hide show
  1. paper_chat_tab.py +2 -1
paper_chat_tab.py CHANGED
@@ -166,7 +166,8 @@ def create_chat_interface(model_name, paper_content, hf_token_input):
166
  if model_name not in tokenizer_cache:
167
  # Load the tokenizer from Hugging Face
168
  # tokenizer = AutoTokenizer.from_pretrained(model_name)
169
- tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct")
 
170
  tokenizer_cache[model_name] = tokenizer
171
  else:
172
  tokenizer = tokenizer_cache[model_name]
 
166
  if model_name not in tokenizer_cache:
167
  # Load the tokenizer from Hugging Face
168
  # tokenizer = AutoTokenizer.from_pretrained(model_name)
169
+ tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct",
170
+ token=os.environ.get("HF_TOKEN"))
171
  tokenizer_cache[model_name] = tokenizer
172
  else:
173
  tokenizer = tokenizer_cache[model_name]