joaogante HF staff commited on
Commit
918037c
1 Parent(s): 980632f

fiddle with plots

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -148,9 +148,15 @@ def get_plot(model_name, plot_eager, generate_type):
148
  df = df[df["framework"] != "TF (Eager Execition)"]
149
 
150
  g = sns.catplot(
151
- data=df, kind="bar",
152
- x="variable", y="value", hue="framework",
153
- ci="sd", palette="dark", alpha=.6, height=6
 
 
 
 
 
 
154
  )
155
  g.despine(left=True)
156
  g.set_axis_labels("GPU", "Generation time (ms)")
 
148
  df = df[df["framework"] != "TF (Eager Execition)"]
149
 
150
  g = sns.catplot(
151
+ data=df,
152
+ kind="bar",
153
+ x="variable",
154
+ y="value",
155
+ hue="framework",
156
+ ci="sd",
157
+ palette={"PyTorch": "blue", "TF (Eager Execition)": "orange", "TF (XLA)": "red"},
158
+ alpha=.6,
159
+ height=6
160
  )
161
  g.despine(left=True)
162
  g.set_axis_labels("GPU", "Generation time (ms)")