upgrades to dependencies and fixing the docs

Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.11-slim
+FROM python:3.12-slim
 
 RUN apt-get update && apt-get install -y \
     build-essential \

README.md

@@ -20,7 +20,7 @@ A live version can be found here: https://huggingface.co/spaces/ProtectAI/llm-gu
 2. Install dependencies (preferably in a virtual environment)
 
 ```sh
-pip install -r requirements.txt
+pip install -r requirements.txt -U
 ```
 
 3. Start the app:

app.py

@@ -20,7 +20,7 @@ st.set_page_config(
     layout="wide",
     initial_sidebar_state="expanded",
     menu_items={
-        "About": "https://llm-guard.com/",
+        "About": "https://protectai.github.io/llm-guard/",
     },
 )
 
@@ -30,7 +30,7 @@ logger.setLevel(logging.INFO)
 # Sidebar
 st.sidebar.header(
     """
-Scanning prompt and output using [LLM Guard](https://llm-guard.com/)
+Scanning prompt and output using [LLM Guard](https://protectai.github.io/llm-guard/)
 """
 )
 
@@ -53,7 +53,7 @@ with st.expander("About", expanded=False):
     st.info(
         """LLM-Guard is a comprehensive tool designed to fortify the security of Large Language Models (LLMs).
         \n\n[Code](https://github.com/protectai/llm-guard) |
-        [Documentation](https://llm-guard.com/)"""
+        [Documentation](https://protectai.github.io/llm-guard/)"""
     )
 
 analyzer_load_state = st.info("Starting LLM Guard...")

output.py

@@ -4,7 +4,7 @@ from datetime import timedelta
 from typing import Dict, List
 
 import streamlit as st
-from llm_guard.input_scanners.anonymize import default_entity_types
+from llm_guard.input_scanners.anonymize import DEFAULT_ENTITY_TYPES
 from llm_guard.input_scanners.code import SUPPORTED_LANGUAGES as SUPPORTED_CODE_LANGUAGES
 from llm_guard.output_scanners import get_scanner_by_name
 from llm_guard.output_scanners.bias import MatchType as BiasMatchType
@@ -47,7 +47,7 @@ def init_settings() -> (List, Dict):
         "Select scanners",
         options=all_scanners,
         default=all_scanners,
-        help="The list can be found here: https://llm-guard.com/output_scanners/bias/",
+        help="The list can be found here: https://protectai.github.io/llm-guard/output_scanners/bias/",
     )
 
     settings = {}
@@ -424,14 +424,14 @@ def init_settings() -> (List, Dict):
             st_sens_entity_types = st_tags(
                 label="Sensitive entities",
                 text="Type and press enter",
-                value=default_entity_types,
-                suggestions=default_entity_types
+                value=DEFAULT_ENTITY_TYPES,
+                suggestions=DEFAULT_ENTITY_TYPES
                 + ["DATE_TIME", "NRP", "LOCATION", "MEDICAL_LICENSE", "US_PASSPORT"],
                 maxtags=30,
                 key="sensitive_entity_types",
             )
             st.caption(
-                "Check all supported entities: https://llm-guard.com/input_scanners/anonymize/"
+                "Check all supported entities: https://protectai.github.io/llm-guard/input_scanners/anonymize/"
             )
             st_sens_redact = st.checkbox("Redact", value=False, key="sens_redact")
             st_sens_threshold = st.slider(

prompt.py

@@ -5,7 +5,7 @@ from typing import Dict, List
 
 import streamlit as st
 from llm_guard.input_scanners import get_scanner_by_name
-from llm_guard.input_scanners.anonymize import default_entity_types
+from llm_guard.input_scanners.anonymize import DEFAULT_ENTITY_TYPES
 from llm_guard.input_scanners.code import SUPPORTED_LANGUAGES as SUPPORTED_CODE_LANGUAGES
 from llm_guard.input_scanners.gibberish import MatchType as GibberishMatchType
 from llm_guard.input_scanners.language import MatchType as LanguageMatchType
@@ -39,7 +39,7 @@ def init_settings() -> (List, Dict):
         "Select scanners",
         options=all_scanners,
         default=all_scanners,
-        help="The list can be found here: https://llm-guard.com/input_scanners/anonymize/",
+        help="The list can be found here: https://protectai.github.io/llm-guard/input_scanners/anonymize/",
     )
 
     settings = {}
@@ -54,14 +54,14 @@ def init_settings() -> (List, Dict):
             st_anon_entity_types = st_tags(
                 label="Anonymize entities",
                 text="Type and press enter",
-                value=default_entity_types,
-                suggestions=default_entity_types
+                value=DEFAULT_ENTITY_TYPES,
+                suggestions=DEFAULT_ENTITY_TYPES
                 + ["DATE_TIME", "NRP", "LOCATION", "MEDICAL_LICENSE", "US_PASSPORT"],
                 maxtags=30,
                 key="anon_entity_types",
             )
             st.caption(
-                "Check all supported entities: https://llm-guard.com/input_scanners/anonymize/"
+                "Check all supported entities: https://protectai.github.io/llm-guard/input_scanners/anonymize/"
             )
             st_anon_hidden_names = st_tags(
                 label="Hidden names to be anonymized",

requirements.txt

@@ -1,5 +1,5 @@
-llm-guard==0.3.12
-llm-guard[onnxruntime]==0.3.12
-pandas==2.2.2
-streamlit==1.33.0
+llm-guard==0.3.16
+llm-guard[onnxruntime]==0.3.16
+pandas==2.3.1
+streamlit==1.46.1
 streamlit-tags==1.2.8