Spaces:
Sleeping
Sleeping
Update services/guardrails.py
Browse files- services/guardrails.py +1 -0
services/guardrails.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
from models import llama_guard_client
|
| 2 |
from config import config
|
| 3 |
import sys
|
|
|
|
| 4 |
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'models'))
|
| 5 |
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'config'))
|
| 6 |
def filter_input_with_llama_guard(user_input: str, model: str = None) -> str:
|
|
|
|
| 1 |
from models import llama_guard_client
|
| 2 |
from config import config
|
| 3 |
import sys
|
| 4 |
+
import os
|
| 5 |
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'models'))
|
| 6 |
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'config'))
|
| 7 |
def filter_input_with_llama_guard(user_input: str, model: str = None) -> str:
|