Spaces:

collinear-ai
/

collinear-guard-demo

Sleeping

tanveeshsingh commited on Oct 15, 2024

Commit

f965215

1 Parent(s): 505e6e2

Change endpoint

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ def llama_guard_classify(conv_prefix, response):
     return output.choices[0].message.content
 def classify_prompt(category,conv_prefix, response):
-    url = "https://api.collinear.ai/api/v1/dataset/"
     payload = {
         "model_name": "collinear_guard_classifier",

     return output.choices[0].message.content
 def classify_prompt(category,conv_prefix, response):
+    url = f"{API_ENDPOINT}/api/v1/judge/safety"
     payload = {
         "model_name": "collinear_guard_classifier",