Spaces:

pratikshahp
/

finance-chatbot-with-guardrails

Sleeping

pratikshahp commited on Dec 11, 2024

Commit

d78ba60

verified ·

1 Parent(s): 9bf6e30

Update guardrail.py

Files changed (1) hide show

guardrail.py CHANGED Viewed

@@ -52,7 +52,7 @@ def is_safe(message: str) -> bool:
     - If unsafe, a second line must include a comma-separated list of violated categories. [/INST]"""
     response = client.completions.create(
-        model="Meta-Llama/LlamaGuard-3-8b",
         prompt=prompt
     )

     - If unsafe, a second line must include a comma-separated list of violated categories. [/INST]"""
     response = client.completions.create(
+        model="meta-llama/Meta-Llama-Guard-3-8B",
         prompt=prompt
     )