Update app.py

app.py

@@ -47,7 +47,7 @@ def mmlu_display_question_answer(question, cot, request: gr.Request):
     else:
         q = mmlu_question_selector_map[question]
 
-    return q["agent_response"]["llama"][0], q["agent_response"]["wizardlm"][0], q["agent_response"]["orca"][0], q["agent_response"]["llama"][1], q["agent_response"]["wizardlm"][1], q["agent_response"]["orca"][1], q["agent_response"]["llama"][2], q["agent_response"]["wizardlm"][2], q["agent_response"]["orca"][2]
+    return q["agent_response"]["llama"][0], q["agent_response"]["wizardlm"][0], q["agent_response"]["orca"][0], q["summarization"][0], q["agent_response"]["llama"][1], q["agent_response"]["wizardlm"][1], q["agent_response"]["orca"][1], q["summarization"][1], q["agent_response"]["llama"][2], q["agent_response"]["wizardlm"][2], q["agent_response"]["orca"][2]
     
 
 def warmup(model_list=model_list, model_inference_endpoints=inference_endpoint):
@@ -240,10 +240,10 @@ with gr.Blocks() as demo:
 
             with gr.Column():
                 with gr.Row(elem_id="model1_response"):
-                    math_model1_output1 = gr.Textbox(label="Llama2🦙's 1️⃣st response")   # value=math_result[math_question_list]["agent_response"]["llama"][0]
+                    math_model1_output1 = gr.Textbox(label="Llama2🦙's 1️⃣st response")
                     math_model2_output1 = gr.Textbox(label="WizardLM🧙‍♂️'s 1️⃣st response")
                     math_model3_output1 = gr.Textbox(label="Orca🐬's 1️⃣st response")
-                math_summarization_text1 = gr.Textbox(lebel="Summarization 1️⃣")
+                math_summarization_text1 = gr.Textbox(label="Summarization 1️⃣")
                 with gr.Row(elem_id="model2_response"):
                     math_model1_output2 = gr.Textbox(label="Llama2🦙's 2️⃣nd response")
                     math_model2_output2 = gr.Textbox(label="WizardLM🧙‍♂️'s 2️⃣nd response")
@@ -260,12 +260,12 @@ with gr.Blocks() as demo:
         math_cot.select(
             math_display_question_answer,
             [math_question_list, math_cot],
-            [math_model1_output1, math_model2_output1, math_model3_output1, math_model1_output2, math_model2_output2, math_model3_output2, math_model1_output3, math_model2_output3, math_model3_output3]
+            [math_model1_output1, math_model2_output1, math_model3_output1, math_summarization_text1, math_model1_output2, math_model2_output2, math_model3_output2, math_summarization_text2, math_model1_output3, math_model2_output3, math_model3_output3]
         )
         math_question_list.change(
             math_display_question_answer,
             [math_question_list, math_cot],
-            [math_model1_output1, math_model2_output1, math_model3_output1, math_model1_output2, math_model2_output2, math_model3_output2, math_model1_output3, math_model2_output3, math_model3_output3]
+            [math_model1_output1, math_model2_output1, math_model3_output1, math_summarization_text1, math_model1_output2, math_model2_output2, math_model3_output2, math_summarization_text2, math_model1_output3, math_model2_output3, math_model3_output3]
         )