* bump to version 0.2.0 of llm-guard

README.md

@@ -1,5 +1,5 @@
 ---
-title: LLM Guard Demo
+title: LLM Guard Playground
 emoji: 🏢
 colorFrom: blue
 colorTo: gray
@@ -11,11 +11,11 @@ license: mit
 Here's a simple app, written in pure Python, to create a demo website for LLM Guard.
 The app is based on the [streamlit](https://streamlit.io/) package.
 
-A live version can be found here: https://huggingface.co/spaces/laiyer/llm-guard-demo
+A live version can be found here: https://huggingface.co/spaces/laiyer/llm-guard-playground
 
 ## Requirements
 
-1. Clone the repo and move to the `examples/demo` folder
+1. Clone the repo and move to the `examples/playground` folder
 
 2. Install dependencies (preferably in a virtual environment)
 

app.py

@@ -21,7 +21,7 @@ OUTPUT = "output"
 vault = Vault()
 
 st.set_page_config(
-    page_title="LLM Guard demo",
+    page_title="LLM Guard Playground",
     layout="wide",
     initial_sidebar_state="expanded",
     menu_items={
@@ -29,7 +29,7 @@ st.set_page_config(
     },
 )
 
-logger = logging.getLogger("llm-guard-demo")
+logger = logging.getLogger("llm-guard-playground")
 logger.setLevel(logging.INFO)
 
 # Sidebar
@@ -49,7 +49,7 @@ elif scanner_type == OUTPUT:
     enabled_scanners, settings = init_output_settings()
 
 # Main pannel
-with st.expander("About this demo", expanded=False):
+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/laiyer-ai/llm-guard) |
@@ -127,7 +127,7 @@ except Exception as e:
 
 # After:
 if st_is_valid is not None:
-    st.subheader(f"Results - {'valid' if st_is_valid else 'invalid'} ({st_time_delta.total_seconds()} seconds)")
+    st.subheader(f"Results - {'valid' if st_is_valid else 'invalid'} ({round(st_time_delta.total_seconds())} seconds)")
 
     col1, col2 = st.columns(2)
 

output.py

@@ -22,7 +22,7 @@ from llm_guard.output_scanners.sentiment import Sentiment
 from llm_guard.output_scanners.toxicity import Toxicity
 from llm_guard.vault import Vault
 
-logger = logging.getLogger("llm-guard-demo")
+logger = logging.getLogger("llm-guard-playground")
 
 
 def init_settings() -> (List, Dict):

prompt.py

@@ -18,7 +18,7 @@ from llm_guard.input_scanners import (
 from llm_guard.input_scanners.anonymize import default_entity_types
 from llm_guard.vault import Vault
 
-logger = logging.getLogger("llm-guard-demo")
+logger = logging.getLogger("llm-guard-playground")
 
 
 def init_settings() -> (List, Dict):

requirements.txt

@@ -1,4 +1,4 @@
-llm-guard==0.1.3
+llm-guard==0.2.0
 pandas==2.1.0
 streamlit==1.26.0
 streamlit-tags==1.2.8