Changes

app.py

@@ -41,7 +41,7 @@ def update_inputs(input_style):
         return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=True)
 
 
-async def lynx(input_style_dropdown,document_input,question_input,answer_input,collinear_output):
+async def lynx(input_style_dropdown,document_input,question_input,answer_input,result_output):
     start_time = time.time()
     if input_style_dropdown=='QA format':
         client = AsyncOpenAI(
@@ -79,12 +79,12 @@ Your output should be in JSON FORMAT with the keys "REASONING" and "SCORE":
     else:
         results = 'NA'
     lynx_time = round(time.time() - start_time, 2)  # Calculate time taken for Lynx
-    await add_to_dataset(input_style_dropdown,document_input,question_input,answer_input,claim_input,results,collinear_output)
+    await add_to_dataset(input_style_dropdown,document_input,question_input,answer_input,claim_input,results,result_output)
     return results, lynx_time
 # Function to judge reliability based on the selected input format
 
 
-async def add_to_dataset(category,document,question,answer,claim,conv_prefix,lynx_output,veritas_output):
+async def add_to_dataset(category,document,question,answer,claim,conv_prefix,lynx_output,result_output):
     conv_prefix = convert_to_message_array(conv_prefix)
     dataset = load_dataset("collinear-ai/veritas-demo-dataset")
     new_row = {
@@ -96,7 +96,7 @@ async def add_to_dataset(category,document,question,answer,claim,conv_prefix,lyn
     'conv_prefix':conv_prefix[:-1],
     'response':conv_prefix[-1],
     'lynx_output':lynx_output,
-    'veritas_output':veritas_output,
+    'veritas_output':result_output,
         }
     train_dataset = dataset['train']