Changes

app.py

@@ -5,6 +5,10 @@ import openai
 API_ENDPOINT = "https://txl0ptjvttfogwt9.us-east-1.aws.endpoints.huggingface.cloud/v1/"
 API_KEY = "NA"
 
+def llama_guard_classify(conv_prefix, response):
+    # Simulating LLaMA-Guard classification, replace with actual implementation
+    return '{"output": 1}'
+
 def classify_prompt(conv_prefix, response):
     template_str = """
 Please review the scoring criteria:
@@ -47,7 +51,8 @@ Now, please output the following as a JSON object:
 def process_inputs(conv_prefix, response_content):
     response = {"role": "assistant", "content": response_content}
     output = classify_prompt(conv_prefix, response)
-    return output
+    llama_output = llama_guard_classify(conv_prefix, response)
+    return output,llama_output
 
 # Gradio Interface
 demo = gr.Interface(
@@ -57,9 +62,9 @@ demo = gr.Interface(
         
         gr.Textbox(lines=2, placeholder="Enter the assistant's response", label="Assistant Response", value='I am good! Thanks for asking')
     ],
-    outputs="text",
-    title="Prompt Safety Classification",
-    description="Classify a conversation prompt's safety by providing a conversation prefix (array of objects) and an assistant's response."
+    outputs=[gr.Textbox(label="OpenAI Output"), gr.Textbox(label="LLaMA-Guard Output"),
+    title="Safety Classifier",
+    description="Classify a conversation's safety by providing a conversation prefix (array of objects) and an assistant's response."
 )
 
 demo.launch()