Afshintm commited on
Commit
d24cfe6
·
verified ·
1 Parent(s): a72cede

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -541,13 +541,13 @@ def agentic_rag(query: str):
541
 
542
 
543
  # Function to filter user input with Llama Guard
544
- def filter_input_with_llama_guard(user_input, model="llama-guard-3-8b"):
545
  """
546
  Filters user input using Llama Guard to ensure it is safe.
547
 
548
  Parameters:
549
  - user_input: The input provided by the user.
550
- - model: The Llama Guard model to be used for filtering (default is "llama-guard-3-8b").
551
 
552
  Returns:
553
  - The filtered and safe input.
 
541
 
542
 
543
  # Function to filter user input with Llama Guard
544
+ def filter_input_with_llama_guard(user_input, model="meta-llama/llama-guard-4-12b"):
545
  """
546
  Filters user input using Llama Guard to ensure it is safe.
547
 
548
  Parameters:
549
  - user_input: The input provided by the user.
550
+ - model: The Llama Guard model to be used for filtering (default is "meta-llama/llama-guard-4-12b").
551
 
552
  Returns:
553
  - The filtered and safe input.