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: