Spaces:
Build error
Build error
Commit
·
e7a0816
1
Parent(s):
e3ebc85
Changes for model
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def classify_prompt(category,conv_prefix, response):
|
|
35 |
url = f"{API_ENDPOINT}/api/v1/judge/safety"
|
36 |
|
37 |
payload = {
|
38 |
-
"model_name": "
|
39 |
"nano_model_type": category,
|
40 |
"conversation": conv_prefix,
|
41 |
"response": response
|
|
|
35 |
url = f"{API_ENDPOINT}/api/v1/judge/safety"
|
36 |
|
37 |
payload = {
|
38 |
+
"model_name": "collinear_guard_classifier_beta",
|
39 |
"nano_model_type": category,
|
40 |
"conversation": conv_prefix,
|
41 |
"response": response
|