Update llama_guard.py

llama_guard.py

@@ -15,7 +15,7 @@ HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACE_LLAMAGAURD_API_TOKEN")
 
 # reading unsafe categories file
 try:
-    with open("data/llama_guard-unsafe_categories.txt", "r") as file:
+    with open("data/llama_guard-unsafe_files.txt", "r") as file:
         unsafe_categories = file.read()
 except FileNotFoundError:
     print("File not found. Please check the file path.")