Update app.py

app.py

@@ -85,8 +85,11 @@ def run_mmlu_evaluation(all_subjects, num_subjects, num_shots, all_questions, nu
         load_model()
 
     if not model_loaded:
-        return "⚠️ Model not loaded. Please load the model first.", None, gr.update(interactive=True), gr.update(visible=False), \
-              [gr.update(interactive=True) for _ in range(5)]
+        return ("⚠️ Model not loaded. Please load the model first.", None, 
+                gr.update(interactive=True), gr.update(visible=False),
+                gr.update(interactive=True), gr.update(interactive=True), 
+                gr.update(interactive=True), gr.update(interactive=True), 
+                gr.update(interactive=True))
 
     # Convert num_subjects to -1 if all_subjects is True
     if all_subjects:
@@ -145,8 +148,11 @@ def run_mmlu_evaluation(all_subjects, num_subjects, num_shots, all_questions, nu
     )
 
     # Return values that re-enable UI components after completion
-    return report, results_df, gr.update(interactive=True), gr.update(visible=False), \
-           [gr.update(interactive=True) for _ in range(5)]
+    return (report, results_df, 
+            gr.update(interactive=True), gr.update(visible=False),
+            gr.update(interactive=True), gr.update(interactive=True), 
+            gr.update(interactive=True), gr.update(interactive=True), 
+            gr.update(interactive=True))
     
 # ---------------------------------------------------------------------------
 # 4. Gradio Interface
@@ -319,9 +325,13 @@ with gr.Blocks() as demo:
         outputs=[
             results_output,
             results_table,
-            eval_mmlu_button,
+            eval_mmlu_button, 
             cancel_mmlu_button,
-            [all_subjects_checkbox, num_subjects_slider, num_shots_slider, all_questions_checkbox, num_questions_slider]
+            all_subjects_checkbox,
+            num_subjects_slider,
+            num_shots_slider,
+            all_questions_checkbox,
+            num_questions_slider
         ]
     )