tomy007 commited on
Commit
568c9c7
·
verified ·
1 Parent(s): d6eec00

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -466,13 +466,13 @@ def agentic_rag(query: str):
466
  #================================ Guardrails ===========================#
467
  llama_guard_client = Groq(api_key=llama_api_key)
468
  # Function to filter user input with Llama Guard
469
- def filter_input_with_llama_guard(user_input, model="llama-guard-3-8b"):
470
  """
471
  Filters user input using Llama Guard to ensure it is safe.
472
 
473
  Parameters:
474
  - user_input: The input provided by the user.
475
- - model: The Llama Guard model to be used for filtering (default is "llama-guard-3-8b").
476
 
477
  Returns:
478
  - The filtered and safe input.
 
466
  #================================ Guardrails ===========================#
467
  llama_guard_client = Groq(api_key=llama_api_key)
468
  # Function to filter user input with Llama Guard
469
+ def filter_input_with_llama_guard(user_input, model="meta-llama/llama-guard-4-12b"):
470
  """
471
  Filters user input using Llama Guard to ensure it is safe.
472
 
473
  Parameters:
474
  - user_input: The input provided by the user.
475
+ - model: The Llama Guard model to be used for filtering (default is "meta-llama/llama-guard-4-12b").
476
 
477
  Returns:
478
  - The filtered and safe input.