* 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")
|