add token

app.py

@@ -276,6 +276,7 @@ en_stopwords = set(
 import multiprocessing
 import os
 
+HF_TOKEN = os.getenv("HUGGING_FACE_HUB_TOKEN")
 import tqdm
 import whoosh.index as whoosh_index
 from whoosh.analysis import StemmingAnalyzer
@@ -347,12 +348,12 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
 safety_tokenizer = tokenizer = AutoTokenizer.from_pretrained(
     "salexashenko/T5-Base-ROT-epoch-2-train-loss-1.3495-val-loss-1.4164",
-    use_auth_token=True,
+    use_auth_token=HF_TOKEN,
 )
 safety_model = model = (
     AutoModelForSeq2SeqLM.from_pretrained(
         "salexashenko/T5-Base-ROT-epoch-2-train-loss-1.3495-val-loss-1.4164",
-        use_auth_token=True,
+        use_auth_token=HF_TOKEN,
     )
     .half()
     .cuda()