Update app.py

app.py

@@ -75,7 +75,10 @@ def get_plots(task):
     fig.update_layout(
         xaxis_title="Model",
         yaxis_title="GPU Energy (Wh)",
-        yaxis_tickformat=".4f" # Add this line to format y-axis ticks
+        yaxis_tickformat=".4f", # Add this line to format y-axis ticks
+        yaxis = dict(
+            tickformat=".4f" # Ensure tickformat is set within yaxis dict as well
+        )
     )
     return fig
 
@@ -113,7 +116,10 @@ def get_all_plots():
     fig.update_layout(
         xaxis_title="Model",
         yaxis_title="GPU Energy (Wh)",
-        yaxis_tickformat=".4f" # Add this line to format y-axis ticks
+        yaxis_tickformat=".4f", # Add this line to format y-axis ticks
+        yaxis = dict(
+            tickformat=".4f" # Ensure tickformat is set within yaxis dict as well
+        )
     )
     return fig
 
@@ -171,6 +177,7 @@ def get_text_generation_plots(model_class):
         width=800,
         color_discrete_map=color_map
     )
+    # Update hover text to show the model and GPU Energy (with 4 decimals)
     fig.update_traces(
         hovertemplate="<br>".join([
             "Model: %{x}",
@@ -181,7 +188,10 @@ def get_text_generation_plots(model_class):
     fig.update_layout(
         xaxis_title="Model",
         yaxis_title="GPU Energy (Wh)",
-        yaxis_tickformat=".4f" # Add this line to format y-axis ticks
+        yaxis_tickformat=".4f", # Add this line to format y-axis ticks
+        yaxis = dict(
+            tickformat=".4f" # Ensure tickformat is set within yaxis dict as well
+        )
     )
     return fig
 
@@ -233,12 +243,12 @@ Select different tasks to see scored models. Submit open models for testing and
             # Dropdown moved above the plot and leaderboard
             model_class_dropdown = gr.Dropdown(choices=["A", "B", "C"],
                                                  label="Select Model Class",
-                                                 value="A")
+                                                 value="C") # Default to C for testing
             with gr.Row():
                 with gr.Column(scale=1.3):
-                    tg_plot = gr.Plot(get_text_generation_plots("A"))
+                    tg_plot = gr.Plot(get_text_generation_plots("C")) # Default to C for testing
                 with gr.Column(scale=1):
-                    tg_table = gr.Dataframe(get_text_generation_model_names("A"), datatype="markdown")
+                    tg_table = gr.Dataframe(get_text_generation_model_names("C"), datatype="markdown")
             # Update plot and table when the dropdown value changes
             model_class_dropdown.change(fn=update_text_generation,
                                         inputs=model_class_dropdown,