Spaces:

joaogante
/

tf_xla_generate_benchmarks

Running

joaogante HF staff commited on Jul 15, 2022

Commit

e3f1090

1 Parent(s): 6d9bae3

now with labels

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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("")
     return plt.gcf()
 demo = gr.Blocks()

         ci="sd", palette="dark", alpha=.6, height=6
     )
     g.despine(left=True)
+    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()