Update app.py

app.py

@@ -143,7 +143,7 @@ with gr.Blocks(css=CSS) as demo:
             ]
         )
     
-    # Connect evaluation button with state tracking
+        # Connect evaluation button with state tracking
     action_components['eval_button'].click(
         fn=start_evaluation,
         inputs=[evaluation_state],
@@ -189,10 +189,11 @@ with gr.Blocks(css=CSS) as demo:
             model_components['model2_shots'],
             model_components['model2_regex'],
             model_components['model2_flash_attn']
-        ]
+        ],
+        outputs=results_components['output']  # Change: Only pass 1 output to match the dictionary from run_mmlu_evaluation
     ).then(
         fn=handle_evaluation_results,
-        inputs=[],  # The result from the previous function is automatically passed
+        inputs=[results_components['output']],  # Change: Pass the dictionary to handle_evaluation_results
         outputs=[
             results_components['output'],
             results_components['table'],