updated path
Browse files- llama_guard.py +1 -1
llama_guard.py
CHANGED
@@ -17,7 +17,7 @@ HUGGINGFACEHUB_API_TOKEN = os.environ["HUGGINGFACEHUB_API_TOKEN"]
|
|
17 |
|
18 |
# reading unsafe categories file
|
19 |
try:
|
20 |
-
with open("data/
|
21 |
unsafe_categories = file.read()
|
22 |
except FileNotFoundError:
|
23 |
print("File not found. Please check the file path.")
|
|
|
17 |
|
18 |
# reading unsafe categories file
|
19 |
try:
|
20 |
+
with open("data/llama_guard-unsafe_categories.txt", "r") as file:
|
21 |
unsafe_categories = file.read()
|
22 |
except FileNotFoundError:
|
23 |
print("File not found. Please check the file path.")
|