Changes

app.py

@@ -64,7 +64,6 @@ Now, please output the following as a JSON object:
         {"role": "system", "content": system_prompt},
         {"role": "user", "content": rendered_message}
     ]
-    print(messages)
     completion = client.chat.completions.create(
         model="tgi",
         messages=messages,
@@ -72,13 +71,12 @@ Now, please output the following as a JSON object:
     )
     output = completion.choices[0].message.content
     output = json.loads(output)
-    print(output)
     if output['output']==1:
         return 'safe'
     else:
         return 'unsafe'
 
-def add_to_dataset(safe_text,unsafe_text,conv_prefix, response,llama_resp,collinear_resp):
+async def add_to_dataset(safe_text,unsafe_text,conv_prefix, response,llama_resp,collinear_resp):
     dataset = load_dataset("collinear-ai/collinear-guard-demo")
     new_row = {'safe_text':safe_text,
             'unsafe_text':unsafe_text,
@@ -129,7 +127,7 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         category = gr.Dropdown(
-            ["response","prompt", "refusal"], label="Select Type", value='prompt'
+            ["response","prompt", "refusal"], label="Select Evaluation Type", value='prompt'
         )
 
     with gr.Row():