quick fix

app.py

@@ -7,7 +7,8 @@ API_KEY = "NA"
 
 LLAMA_API_ENDPOINT=os.getenv("LLAMA_API_ENDPOINT")
 LLAMA_API_KEY=os.getenv("LLAMA_API_KEY")
-def llama_guard_classify(conv_prefix, response):
+def llama_guard_classify(conv_prefix, response_content):
+    response = {"role": "assistant", "content": response_content}
     model_name = 'meta-llama/Meta-Llama-Guard-3-8B'
     client = openai.OpenAI(
         base_url=LLAMA_API_ENDPOINT,
@@ -74,7 +75,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="OpenAI Output"), gr.Textbox(label="LLaMA-Guard Output")],
+    outputs=[gr.Textbox(label="Collinear Guard 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."
 )