Update app.py

app.py

@@ -40,6 +40,13 @@ def plot_acc_param(param_compare_results_df: pl.DataFrame, width: int = 1000, he
 
 
 def plot_acc_memory(memory_compare_results_df: pl.DataFrame, width: int = 900, height: int = 640) -> alt.LayerChart:
+    if len(memory_compare_results_df) > 0:
+        batch_size = memory_compare_results_df["batch_size"][0]
+        amp = memory_compare_results_df["amp"][0]
+    else:
+        batch_size = ""
+        amp = ""
+
     df = memory_compare_results_df.select(
         "Model name",
         "Model type",
@@ -71,7 +78,9 @@ def plot_acc_memory(memory_compare_results_df: pl.DataFrame, width: int = 900, h
     )
 
     chart = base + text + frontier
-    return chart.properties(title="Accuray vs GPU Memory", width=width, height=height).configure_scale(zero=False)
+    return chart.properties(
+        title=f"Accuray vs GPU Memory (batch size={batch_size}, amp={amp}", width=width, height=height
+    ).configure_scale(zero=False)
 
 
 def plot_acc_rate(rate_compare_results_df: pl.DataFrame, width: int = 1000, height: int = 680) -> alt.LayerChart: