now with labels

app.py

@@ -151,8 +151,8 @@ def get_plot(model_name, generate_type):
         ci="sd", palette="dark", alpha=.6, height=6
     )
     g.despine(left=True)
-    # g.set_axis_labels("", "Body mass (g)")
-    # g.legend.set_title("")
+    g.set_axis_labels("GPU", "Generation time (ms)")
+    g.legend.set_title(f"{generate_type} generation time for {model_name}")
     return plt.gcf()
 
 demo = gr.Blocks()