bayartsogt commited on
Commit
a363f8d
1 Parent(s): 0afb4f9

add HF_TOKEN

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,9 +1,12 @@
 
1
  from typing import List
2
  import gradio as gr
3
  from transformers import AutoTokenizer
4
  import tiktoken
5
 
6
 
 
 
7
  hf_tokenizer_list = [
8
  ("tugstugi/bert-large-mongolian-cased", False),
9
  ("tugstugi/bert-large-mongolian-uncased", False),
@@ -19,7 +22,7 @@ openai_tokenizer_list = [
19
 
20
  # load tokenizers
21
  hf_tokenizers = [
22
- AutoTokenizer.from_pretrained(model_name_or_id, use_fast=use_fast, trust_remote_code=True)
23
  for model_name_or_id, use_fast in hf_tokenizer_list
24
  ]
25
 
 
1
+ import os
2
  from typing import List
3
  import gradio as gr
4
  from transformers import AutoTokenizer
5
  import tiktoken
6
 
7
 
8
+ HF_TOKEN = os.getenv("HF_TOKEN")
9
+
10
  hf_tokenizer_list = [
11
  ("tugstugi/bert-large-mongolian-cased", False),
12
  ("tugstugi/bert-large-mongolian-uncased", False),
 
22
 
23
  # load tokenizers
24
  hf_tokenizers = [
25
+ AutoTokenizer.from_pretrained(model_name_or_id, use_fast=use_fast, trust_remote_code=True, token=HF_TOKEN)
26
  for model_name_or_id, use_fast in hf_tokenizer_list
27
  ]
28