Changes for HF Demo

app.py

@@ -42,7 +42,7 @@ def classify_prompt(category,conv_prefix, response):
         judge_id='7fd02b72-655b-4992-9380-ba496eefe12a'
 
     payload = {
-        "model_name": "collinear_guard_classifier_beta",
+        "model_name": "collinear_guard_classifier",
         "nano_model_type": category,
         "conversation": conv_prefix,
         "response": response,
@@ -55,7 +55,7 @@ def classify_prompt(category,conv_prefix, response):
     }
 
     response = requests.request("POST", url, json=payload, headers=headers)
-    print(response.json())
+    print(response)
 
     result = response.json()
     judgement = result['judgement']