* token to store in the dataset
Browse files
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")
|