tanveeshsingh commited on
Commit
e7a0816
·
1 Parent(s): e3ebc85

Changes for model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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": "collinear_guard_classifier",
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