asofter commited on
Commit
7780fae
1 Parent(s): aaf32c1

* token to store in the dataset

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ from transformers import AutoTokenizer, pipeline
21
  logging.basicConfig(level=logging.INFO)
22
  logger = logging.getLogger(__name__)
23
 
24
- hf_api = HfApi()
25
  num_processes = 2 # mp.cpu_count()
26
 
27
  lakera_api_key = os.getenv("LAKERA_API_KEY")
 
21
  logging.basicConfig(level=logging.INFO)
22
  logger = logging.getLogger(__name__)
23
 
24
+ hf_api = HfApi(token=os.getenv("HF_TOKEN"))
25
  num_processes = 2 # mp.cpu_count()
26
 
27
  lakera_api_key = os.getenv("LAKERA_API_KEY")