Update app.py

app.py

@@ -166,7 +166,7 @@ with gr.Blocks(title="Model Test Results Dashboard", css=load_css(), js=js_func)
         with gr.Column(scale=4, elem_classes=["main-content"]):
             with gr.Column(visible=True, elem_classes=["current-view"]) as current_view:
                 summary_display = gr.Plot(
-                    value=create_summary_page(Ci_results.df, Ci_results.available_models, COLORS=COLORS),
+                    value=create_summary_page(Ci_results.df, Ci_results.available_models),
                     label="",
                     format="png",
                     elem_classes=["plot-container"],
@@ -320,7 +320,7 @@ with gr.Blocks(title="Model Test Results Dashboard", css=load_css(), js=js_func)
                 False,
             )
         else:
-            fig = create_summary_page(Ci_results.df, Ci_results.available_models, COLORS=COLORS)
+            fig = create_summary_page(Ci_results.df, Ci_results.available_models)
             return (
                 description,
                 links,
@@ -464,7 +464,7 @@ with gr.Blocks(title="Model Test Results Dashboard", css=load_css(), js=js_func)
             )
         else:
             if last_selected_model and Ci_results.df is not None and not Ci_results.df.empty and last_selected_model in Ci_results.df.index:
-                fig, amd_txt, nvidia_txt = plot_model_stats(Ci_results.df, last_selected_model, COLORS=COLORS)
+                fig, amd_txt, nvidia_txt = plot_model_stats(Ci_results.df, last_selected_model)
                 return (
                     gr.update(visible=True),
                     gr.update(visible=False),
@@ -482,7 +482,7 @@ with gr.Blocks(title="Model Test Results Dashboard", css=load_css(), js=js_func)
                     True,
                 )
             else:
-                fig = create_summary_page(Ci_results.df, Ci_results.available_models, COLORS=COLORS)
+                fig = create_summary_page(Ci_results.df, Ci_results.available_models)
                 return (
                     gr.update(visible=True),
                     gr.update(visible=False),
@@ -549,7 +549,7 @@ with gr.Blocks(title="Model Test Results Dashboard", css=load_css(), js=js_func)
                 gr.update(visible=True),
             selected_model, True)
         else:
-            fig, amd_txt, nvidia_txt = plot_model_stats(Ci_results.df, selected_model, COLORS=COLORS)
+            fig, amd_txt, nvidia_txt = plot_model_stats(Ci_results.df, selected_model)
             return (
                 fig,
                 amd_txt,