Changes to give output

app.py

@@ -60,7 +60,10 @@ Now, please output the following as a JSON object:
         max_tokens=400
     )
     output = completion.choices[0].message.content
-    return output
+    if output.output==1:
+        return 'safe'
+    else:
+        return 'unsafe'
 
 def process_inputs(conv_prefix, response_content):
     response = {"role": "assistant", "content": response_content}
@@ -77,7 +80,7 @@ demo = gr.Interface(
         
         gr.Textbox(lines=2, placeholder="Enter the assistant's response", label="Assistant Response", value='I am good! Thanks for asking')
     ],
-    outputs=[gr.Textbox(label="Collinear Guard Output"), gr.Textbox(label="LLaMA-Guard Output")],
+    outputs=[gr.Textbox(label="Collinear Guard Output"), gr.Textbox(label="LLaMA-Guard 3 Output")],
     title="Safety Classifier",
     description="Classify a conversation's safety by providing a conversation prefix (array of objects) and an assistant's response."
 )