Changes

app.py

@@ -26,7 +26,7 @@ def llama_guard_classify(conv_prefix, response):
  ) 
  return output.choices[0].message.content
 
-def classify_prompt(category,safe_text,unsafe_text,conv_prefix, response):
+def classify_prompt(category,conv_prefix, response):
  url = "https://api.collinear.ai/api/v1/dataset/"
 
  payload = {
@@ -75,10 +75,10 @@ async def add_to_dataset(safe_text,unsafe_text,conv_prefix, response,llama_resp,
 
 
 
-def process_inputs(category,safe_text,unsafe_text,conv_prefix, response_content):
+def process_inputs(category,conv_prefix, response_content):
  response = {"role": "assistant", "content": response_content}
  conv_prefix = json.loads(conv_prefix)
- output = classify_prompt(category,safe_text,unsafe_text,conv_prefix, response)
+ output = classify_prompt(category,conv_prefix, response)
  if category=='response':
  llama_output = llama_guard_classify(conv_prefix, response)
  else: