Upload app.py with huggingface_hub
Browse files
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-
|
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-
|
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.
|