ASledziewska
commited on
Commit
•
a6a602a
1
Parent(s):
e6c8f43
updated API key reference
Browse files- llama_guard.py +1 -3
llama_guard.py
CHANGED
@@ -2,7 +2,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
2 |
import torch
|
3 |
|
4 |
# Docs:- https://huggingface.co/meta-llama/LlamaGuard-7b
|
5 |
-
from dotenv import load_dotenv, find_dotenv
|
6 |
import os
|
7 |
import requests
|
8 |
|
@@ -10,8 +9,7 @@ model_id = "meta-llama/LlamaGuard-7b"
|
|
10 |
|
11 |
|
12 |
# Please ensure you have a .env file available with 'HUGGINGFACEHUB_API_TOKEN'
|
13 |
-
|
14 |
-
HUGGINGFACEHUB_API_TOKEN = os.environ["HUGGINGFACEHUB_API_TOKEN"]
|
15 |
|
16 |
# updated on March 24th
|
17 |
|
|
|
2 |
import torch
|
3 |
|
4 |
# Docs:- https://huggingface.co/meta-llama/LlamaGuard-7b
|
|
|
5 |
import os
|
6 |
import requests
|
7 |
|
|
|
9 |
|
10 |
|
11 |
# Please ensure you have a .env file available with 'HUGGINGFACEHUB_API_TOKEN'
|
12 |
+
HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACE_LLAMAGAURD_API_TOKEN")
|
|
|
13 |
|
14 |
# updated on March 24th
|
15 |
|